pytorch中的tensor.matmul(tensor.T)和tensor @ tensor.T方法

除了tensor.matmul(tensor.T)这种表示方法,还有tensor @ tensor.T,都可以实现张量的矩阵乘法,当然tensor.mm也是矩阵乘法(这些方法和对应元素相乘区分开)
对应元素相乘
1.初始化张量

tensor = torch.ones(4, 4)
tensor[:,1] = 0
print(tensor)
tensor([[1., 0., 1., 1.],
        [1., 0., 1., 1.],
        [1., 0., 1., 1.],
        [1., 0., 1., 1.]])

2.执行矩阵方法

print(f"tensor.matmul(tensor.T) \n {tensor.matmul(tensor.T)} \n")
# Alternative syntax:
print(f"tensor @ tensor.T \n {tensor @ tensor.T}")
tensor.matmul(tensor.T)
 tensor([[3., 3., 3., 3.],
        [3., 3., 3., 3.],
        [3., 3., 3., 3.],
        [3., 3., 3., 3.]])

tensor @ tensor.T
 tensor([[3., 3., 3., 3.],
        [3., 3., 3., 3.],
        [3., 3., 3., 3.],
        [3., 3., 3., 3.]])
posted @ 2021-05-18 16:28  小Aer  阅读(16)  评论(0编辑  收藏  举报  来源