torch.tensor的乘法运算

说明：
这里的tensor不涉及高维张量，即为一维张量或二维张量。

1、a*b

1.1、a和b同行同列

张量中各元素对应相乘。

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

输出：
tensor([[ 1,  4],
        [ 4,  9],
        [ 9, 16]])

2、torch.mm(a,b)

数学中的矩阵相乘。

a = torch.tensor([[1,2],[2,3],[3,4]])
b = torch.tensor([[1,2,3,4],[1,2,3,4]])
print(torch.mm(a,b))

输出：
tensor([[ 3,  6,  9, 12],
        [ 5, 10, 15, 20],
        [ 7, 14, 21, 28]])