PyTorch学习数据格式转换

Posted 2021-06-07 My heart will go ~~

Numpy:用于处理数据的模块。
Torch：是神经网络中的numpy，是一个tensor的形式。
两者之间的对比以及一些简单计算功能代码如下：

数据类型转换

np_data = np.arange(6).reshape((2,3))
torch_data=torch.from_numpy(np_data)#np类型转换成Torch类型
tensor2array=torch_data.numpy()#torch类型转换成numpy类型

data=[-1,-2,1,2]
tensor=torch.FloatTensor(data)#转换成Float类型
a=np.abs(data)#求绝对值
b=torch.abs(tensor)#求绝对值
c=np.sin(a)#求正弦
d=np.mean(c)#求平均

data1=[[1,2],[3,4]]
tensor=torch.FloatTensor(data1)#都转换成Tensor的形式
c=np.matmul(data1,data1)#np里面矩阵相乘
d=torch.mm(tensor,tensor)#torch里面矩阵相乘