案例:
# 矩阵相乘有以下五种方式
a = torch.ones(2,1)
b = torch.ones(1,2)
c = torch.matmul(a, b)
print(c)
print(torch.mm(a, b))
print(a @ b)
print(a.matmul(b))
print(a.mm(b))
"""
tensor([[1., 1.],
[1., 1.]])
tensor([[1., 1.],
[1., 1.]])
tensor([[1., 1.],
[1., 1.]])
tensor([[1., 1.],
[1., 1.]])
tensor([[1., 1.],
[1., 1.]])
"""