数据类型转换

数据类型转换:

1、list 转 numpy

ndarray = np.array(list)

2、numpy转 list

list = ndarray.tolist()

3、 list 转 torch.Tensor

tensor=torch.Tensor(list)

4、 torch.Tensor转 list

#先转numpy，后转list
list=tensor.numpy().tolist()

5、 torch.Tensor 转 numpy

ndarray = tensor.numpy()
​
#gpu上的tensor不能直接转为numpy
​
ndarray = tensor.cpu().numpy()

6、 numpy 转 torch.Tensor

tensor = torch.from_numpy(ndarray)