乘以 PyTorch 张量的所有元素

Question

我想在 PyTorch 中做类似这个问题的事情，即将所有元素相互乘以保持某个轴恒定。PyTorch 是否有任何预定义的功能？

Answer 1

是的。火炬.prod。使用该dim参数来告诉您希望沿着哪个轴计算乘积。

x = torch.randn((2, 2))
print(x)
print(torch.prod(x, 0)) # product along 0th axis

这打印

tensor([[-0.3661, 1.0693],
           [0.5144, 1.3489]])
tensor([-0.1883, 1.4424])