[Pytorch]：PyTorch中张量乘法大全

矩阵乘法：@ortorch.mm()ortorch.matmul()两个矩阵相乘，第一个矩阵的列数必须等于第二个矩阵的行数。

import torch

A = torch.tensor([[1, 2], [3, 4]])
B = torch.tensor([[2, 3], [4, 5]])

result = torch.matmul(A, B)
print(result)

输出：

tensor([[10, 13],
        [22, 29]])

或者使用 @ 运算符执行矩阵乘法：

result = A @ B
print(result)