Pytorch中的矩阵相乘用法

案例：

# 矩阵相乘有以下五种方式
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.]])
"""