pytorch

import torch
import numpy as np
np_data =np.arange(6).reshape((2,3)) #arrange 用于生成一维数组，reshape()将一维数组转换为多维数组
torch_data=torch.from_numpy(np_data) #将numpy数据转换成tensor
tensor2arry =torch_data.numpy()#将tensor 数据转换成numpy
print(’\n numpy’,np_data,
‘\ntorch’,torch_data,
‘\ntensor2arry’,tensor2arry,
)

import torch
import numpy as np
data =[[1,2],[3,4]]
tensor =torch.FloatTensor(data)#将矩阵转换成torch tensor
print('\n numpy:',np.matmul(data,data), #矩阵相乘
      '\ntorch:',torch.mm(tensor,tensor) #矩阵相乘
      )