1.1 list 转 numpy
ndarray = np.array(list)
1.2 numpy 转 list
list = ndarray.tolist()
2.1 list 转 torch.Tensor
tensor=torch.Tensor(list)
2.2 torch.Tensor 转 list
先转numpy,后转list
list = tensor.numpy().tolist()
3.1 torch.Tensor 转 numpy
ndarray = tensor.numpy()
*gpu上的tensor不能直接转为numpy
ndarray = tensor.cpu().numpy()
3.2 numpy 转 torch.Tensor
tensor = torch.from_numpy(ndarray)
https://www.cnblogs.com/siyuan1998/p/10792481.html
该博客主要介绍Python中数据类型的转换方法,包括list与numpy、list与torch.Tensor、torch.Tensor与numpy之间的相互转换,还特别指出gpu上的tensor不能直接转为numpy,需先转到cpu上。
974

被折叠的 条评论
为什么被折叠?



