【python】list,tensor,numpy相互转化

复制代码
# 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)
复制代码

 

posted @   鱼儿冒个泡  阅读(917)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
· 25岁的心里话
点击右上角即可分享
微信分享提示