pytorch学习

import torch
import numpy as np
np_data=np.arange(6).reshape(2,3)
torch_data=torch.from_numpy(np_data)
tensor2array=torch_data.numpy()
data=[-1,-1,2,-2]
tensor=torch.FloatTensor(data)
data=[[1,2],[3,4]]
tensor=torch.FloatTensor(data)
print(
    'numpy',np_data,
    '\ntorch',torch_data,
    '\ntensor2array',tensor2array
)
print('\n',torch.mean(tensor))
print('\nnumpy',np.matmul(data,data),
      '\ntorch',torch.mm(tensor,tensor))