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

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 。