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

Question

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

对于Google员工来说，这是一种产品，也称为：

• 哈达玛产品
• Schur产品
• 整体产品

Answer 1

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

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

注意：对于矩阵乘法，您想使用A @ B，它等效于torch.matmul()。