乘以Pythorch十的所有元素

Question

我想在PyTorch中做这样的事情question，也就是说，将所有元素相乘，保持某个轴不变。Pythorch对此是否有任何预定义的函数？在

Answer 1

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

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

这个指纹

^{pr2}$

Answer 2

假设您想这样做：
矩阵乘法，你可以用火炬.马特姆
将矩阵乘以一个标量（或用标量表示的张量）火炬。倍增

关于这个问题，tensor文档非常广泛。。。你应该看看