【Pytorch】tensor、ndarray、list互相转换方法

最新推荐文章于 2024-03-09 16:30:00 发布

小丫么小阿豪

最新推荐文章于 2024-03-09 16:30:00 发布

阅读量2.2k

点赞数 3

分类专栏： Pytorch 文章标签： python pytorch 计算机视觉人工智能深度学习

本文链接：https://blog.csdn.net/qq_43426078/article/details/124024825

版权

Pytorch 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

import numpy as np
import torch


ndarray = np.array(list)  # list 转 numpy数组
list = ndarray.tolist()  # ndarray 转 list
tensor=torch.Tensor(list)  # list 转 torch.Tensor
list = tensor.numpy().tolist()  # torch.Tensor 转 list 先转ndarray，后转list
ndarray = tensor.cpu().numpy()  # torch.Tensor 转 ndarray （gpu上的tensor需要先copy到CPU中才可以转为numpy数组）
tensor = torch.from_numpy(ndarray)  # ndarray 转 torch.Tensor

优惠劵

小丫么小阿豪

关注关注

3
点赞
踩
9

收藏

觉得还不错? 一键收藏
打赏
0
评论
【Pytorch】tensor、ndarray、list互相转换方法

import numpy as npimport torchndarray = np.array(list) # list 转 numpy数组list = ndarray.tolist() # ndarray 转 listtensor=torch.Tensor(list) # list 转 torch.Tensorlist = tensor.numpy().tolist() # torch.Tensor 转 list 先转ndarray，后转listndarray = tensor.
复制链接

扫一扫