@指的是矩阵乘法,类似于内积,
*指的是pixel-wise乘法。
举个例子,两个3x2的矩阵
>>> a1 = torch.Tensor([[1,2],[3,4], [3, 2]])
>>> a2 = torch.Tensor([[2,3],[1,5], [1, 2]])
矩阵乘法要求第一个矩阵的列=第二个矩阵的行,两个3x2的矩阵直接乘会报错,验证一下
>>> a1@a2
RuntimeError: mat1 and mat2 shapes cannot be multiplied (3x2 and 3x2)
而pixel-wise可以乘
a1*a2
Out:
tensor([[ 2., 6.],
[ 3., 20.],
[ 3., 4.]])
把a1转置成2x3,再相乘,2x3与3x2相乘,结果应该是2x2,验证一下
a1.t()@a2
Out:
tensor([[ 8., 24.],
[10., 30.]])