pytorch中tensor * tensor和tensor.mul(tensor)方法

这两个方法都是对应位置元素相乘,因此要求维度相同
矩阵相乘

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.]])
z1 = tensor * tensor
z2 = tensor.mul(tensor)
print(z1)
print(z2)
tensor([[1., 0., 1., 1.],
        [1., 0., 1., 1.],
        [1., 0., 1., 1.],
        [1., 0., 1., 1.]])
tensor([[1., 0., 1., 1.],
        [1., 0., 1., 1.],
        [1., 0., 1., 1.],
        [1., 0., 1., 1.]])
posted @ 2021-05-18 21:33  小Aer  阅读(5)  评论(0编辑  收藏  举报  来源