pytorch学习

Posted 2021-02-18 -xuewuzhijing-

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,
    ‘
torch‘,torch_data,
    ‘
tensor2array‘,tensor2array
)
print(‘
‘,torch.mean(tensor))
print(‘
numpy‘,np.matmul(data,data),
      ‘
torch‘,torch.mm(tensor,tensor))