博客


torch.prod(input, *, dtype=None) → Tensor


返回

input

张量中所有元素的乘积。

torch.prod(input, dim, keepdim=False, *, dtype=None) → Tensor

返回输入张量给定维度上每行的积

。 输出形状与输入相同，除了给定维度上为1.

input=torch.randn(4,3,2)

out1=torch.prod(input, dim=2)
print(out1)

如果

keepdim

为

True

，则输出张量的大小与

input

大小相同，但

尺寸为1的

dim

除外

。否则，

将

dim

压缩（请参见

torch.squeeze()

），从而使输出张量减少1个维度

。

• 
  input
  
  （
  
  Tensor
  
  ）–输入张量。
• 
  dim
  
  （
  
  int
  
  ）–缩小的尺寸。
• 
  keepdim
  
  （
  
  bool
  
  ）–输出张量是否保持
  
dim

  。