torch 乘法

1、对应元素相乘

a = torch.tensor([[1, 2], [2, 3]])
b = torch.tensor([[1, 2], [2, 3]])
print(a*b)
'''
tensor([[1, 4],
        [4, 9]])
'''
a = torch.tensor([[1, 2], [2, 3]])
b = torch.tensor([[[1, 2], [2, 3]],[[1, 2], [1, 2]]])
print(b.shape)
print(a*b)

2、 数学里的矩阵乘法

torch.mm(a, b)
torch.matmul(a, b)