一、numpy和torch
numpy是python中处理数据的模块,可以处理各种的矩阵(matrix)。
Torch自称为神经网络中的numpy。它会将torch产生的tensor放在GPU中加速运算,就像numpy会把array放在CPU中加速运算。
二.numpy array与torch tensor之间的相互转换
array2tensor=torch.from_numpy(numpy_data)#numpy array->torch tensor,其参数必须是数组形式
tensor2array=torch_data.numpy()#torch tensor->numpy array
三. 用法对比
numpy 数据形式下矩阵a,位置变化用a.transpose(1,2,0)
tensor 数据格式下的矩阵b, 位置变化时 b.permute(1,2,0)