在 PyTorch 中,有多种方法可以执行张量之间的乘法。这里列出了一些常见的乘法操作:
总结:
- 逐元素乘法:
*
ortorch.mul()
- 矩阵乘法:
@
ortorch.mm()
ortorch.matmul()
- 点积:
torch.Tensor.dot()
- 批量矩阵乘法:
torch.bmm()
或torch.matmul()
- 矩阵与向量相乘:
torch.mv(X, w0)
-
逐元素乘法(Element-wise multiplication):
*
ortorch.mul()
()`对应位置的元素相乘,输入张量形状必须相同或可广播。import torch A = torch.tensor([[1, 2], [3, 4]]) B = torch.tensor([[2, 3], [4, 5