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

Question

在 PyTorch 中，如何获得两个向量/矩阵/张量的元素乘积？

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

• 哈达玛产品
• 舒尔产品
• 入门产品

Answer 1

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

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

注意：对于矩阵乘法，您要使用A @ Bwhich 等效于torch.matmul().