乘以PyTorch张量的所有元素

Question

我想在PyTorch中执行类似question的操作，即将所有元素彼此相乘以保持某个轴恒定。 PyTorch是否为此具有任何预定义功能？

Answer 1

是的。 torch.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])

Answer 2

假设您想做：
  矩阵乘法，可以使用torch.matmul
  您可以使用torch.multiply

将矩阵乘以标量（或张量与标量）相乘。

tensor文档在这方面非常广泛...您应该看看