1.numpy与torch tensor类型转换
np_data=np.arange(6).reshape(2,3)
# numpy类型转torch
torch_data=torch.from_numpy(np_data)
#torch转回numpy
tensor2array=torch_data.numpy()
2.numpy list
np_data=np.arange(6).reshape(2,3)
#np转list
list_data=np_data.tolist()
#list转np
np_data2=numpy.array(list_data)
3.pytorch tensor list
list_data=[[0,1,2],[3,4,5]]
#list转pytorch tensor
torch_data=torch.tensor(list_data)
#pytorch tensor转list
list_data2=torch_data.tolist()
感觉好像什么其他类型转list都是tolist()