pytorch 矩阵相乘

可以使用 torch.mm 或 torch.matmul 函数来完成矩阵相乘，在 pytorch 中，二者的作用是相同的，它们都能够计算一个矩阵乘以另一个矩阵后得到的结果，需要注意的是两个矩阵的维度必须匹配。比如：

import torch

a = torch.randn(3, 4)
b = torch.randn(4, 5)

c = torch.mm(a, b)
# 或者 c = torch.matmul(a, b)

print(c)

这里的 a 是一个 3x4 的矩阵，b 是一个 4x5 的矩阵，通过 torch.mm 或 torch.matmul 函数将它们相乘后得到的结果是一个 3x5 的矩阵。