torch 矩阵相乘

import torch
import numpy as np

data = [[1, 2], [3,4]]
tensor = torch.FloatTensor(data)

print(
    '\nnumpy: ',np.matmul(data,data),
    '\ntorch: ',torch.mm(tensor,tensor)
)

numpy:  [[ 7 10]
 [15 22]] 
torch:  
  7  10
 15  22
[torch.FloatTensor of size 2x2]