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.]])