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。