PyTorch 向量/矩阵/张量的元素乘积

Question

在 PyTorch 中，如何获得两个向量/矩阵/张量的element-wise product？

对于谷歌人来说，这个产品也被称为：

• Hadamard 产品
• 舒尔产品
• 入门产品

Answer 1

给定两个张量 A 和 B，您可以使用任一：

• A * B
• torch.mul(A, B)
• A.mul(B)

注意：对于矩阵乘法，你要使用A @ B，它等价于torch.matmul()。