pytorch中tensor、list、numpy之间的转换

最新推荐文章于 2024-08-20 12:41:32 发布

程序员刚子

最新推荐文章于 2024-08-20 12:41:32 发布

阅读量5k

点赞数 12

分类专栏： pytorch 文章标签： pytorch numpy list torch与numpy转换

本文链接：https://blog.csdn.net/qq_33188180/article/details/112840813

版权

7 篇文章 2 订阅

订阅专栏

本文详细介绍如何在Python中进行各种数据类型的转换，包括列表(List)与张量(Tensor)、NumPy数组(NDArray)之间的相互转换，以及这些数据类型与列表之间的转换方法。特别关注了GPU上的Tensor转换为NumPy数组的特殊情况。

摘要由CSDN通过智能技术生成

list转tensor

tensor=torch.Tensor(list)

ndarray = np.array(list)

list = ndarray.tolist()

tensor = torch.from_numpy(ndarray)

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

ndarray = tensor.numpy()
# *gpu上的tensor不能直接转为numpy,应先放回cpu中
ndarray = tensor.cpu().numpy()

CPU ---> GPU: data.cuda()

GPU ---> CPU: data.cpu()

如果tensor是标量的话，可以直接使用 item() 函数（只能是标量）将值取出来：

print(data.item())

关注

专栏目录