Python – list/numpy/pytorch tensor相互转换
1 list与numpy相互转换
- list转numpy
import numpy as np
if __name__ == '__main__':
a = [1,2,3,4]
n = np.array(a)
print(n)
- numpy转list
import numpy as np
if __name__ == '__main__':
n = np.arange(5)
# 第一种方法
#a = n.tolist()
# 第二种方法
a = list(n)
print(a)
list与tensor相互转换
- list转tensor
import numpy as np
import torch
if __name__ == '__main__':
a = [1,2,3,4]
t = torch.tensor(a)
print(t)
- tensor转list:
先转numpy再转list
import numpy as np
import torch
if __name__ == '__main__':
t = torch.arange(5)
a = t.numpy().tolist()
print(a)
numpy与pytorch tensor相互转换
- numpy转tensor
import numpy as np
import torch
if __name__ == '__main__':
n = np.arange(5)
t = torch.from_numpy(n)
print(t)
- tensor转numpy
import numpy as np
import torch
if __name__ == '__main__':
t = torch.arange(5)
n = t.numpy()
print(n)
本文作者:StubbornHuang
版权声明:本文为站长原创文章,如果转载请注明原文链接!
原文标题:Python – list/numpy/pytorch tensor相互转换
原文链接:https://www.stubbornhuang.com/1459/
发布于:2021年07月26日 14:43:02
修改于:2023年06月26日 21:26:08
当前分类随机文章推荐
- Python - 运算符/、//和%的含义与区别 阅读697次,点赞0次
- Python - 获取当前py脚本文件所在的目录路径 阅读972次,点赞0次
- Python – 解决opencv-python使用cv2.imwrite()保存中文路径图片失败的问题 阅读2283次,点赞0次
- Python - 不定长函数参数列表 阅读2371次,点赞0次
- Python - 解决opencv-python使用cv2.imread()读取中文路径图片失败的问题 阅读2152次,点赞1次
- Python - ImportError urllib3 v2.0 only supports OpenSSL 1.1.1+, currently the ‘ssl‘ 阅读20次,点赞0次
- Python - 使用websockets库构建websocket服务器 阅读1894次,点赞0次
- Python - 使用letter box方法缩放图片,防止图片缩放时失真 阅读394次,点赞0次
- Python - 使用python-opencv裁剪原视频为与视频高同宽的视频 阅读1817次,点赞0次
- 简单粗暴:使用pycharm安装对应的Python版本第三方包 阅读3939次,点赞0次
全站随机文章推荐
- 资源分享 - Computer Graphics Principles and Practice (3rd edition) 英文高清PDF下载 阅读3501次,点赞0次
- 资源分享 - Jim Blinn's Corner - Notation, Notation, Notation 英文高清PDF下载 阅读2365次,点赞0次
- 资源分享 - 实时碰撞检测算法技术,Real-Time Collision Detection中文版 PDF下载 阅读2043次,点赞0次
- 资源分享 - Computer Graphics from Scratch - A Programmer's Introduction to 3D Rendering 英文高清PDF下载 阅读1984次,点赞0次
- 资源分享 - 计算机动画算法与技术,Computer Animation - Algorithms and Techniques(Second Edition)中文版PDF下载 阅读1839次,点赞0次
- 资源分享 - Matrix Transforms for Computer Games and Animation 英文PDF下载 阅读950次,点赞0次
- 资源分享 - Fundamentals of Computer Graphics, Third Edition高清英文PDF下载 阅读4607次,点赞1次
- C++ - 日志库easylogging++初始化时不生成默认日志文件mylog.txt 阅读84次,点赞0次
- C++ - std::unordered_map中使用结构体或者vector等复杂的数据结构作为Key值 阅读1220次,点赞0次
- 资源分享 - Graphics Gems II 英文高清PDF下载 阅读2424次,点赞0次
评论
169