pytorch @和*运算

@指的是矩阵乘法,类似于内积,
*指的是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.]])

你可能感兴趣的:(pytorch,java,计算机视觉,pytorch)