Python – list/numpy/pytorch tensor相互转换
本文作者:StubbornHuang
版权声明:本文为站长原创文章,如果转载请注明原文链接!
原文标题:Python – list/numpy/pytorch tensor相互转换
原文链接:https://www.stubbornhuang.com/1459/
发布于:2021年07月26日 14:43:02
修改于:2021年07月26日 14:43:02

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)
当前分类随机文章推荐
- Python - 列表list遍历方法总结 阅读341次,点赞0次
- Python - 使用scikit-video库获取视频的旋转角度并使用opencv_python根据旋转角度对视频进行旋转复原 阅读2452次,点赞1次
- Python – 解决opencv-python使用cv2.imwrite()保存中文路径图片失败的问题 阅读928次,点赞0次
- Python3爬虫 - requests的请求响应状态码(requests.status_code) 阅读8565次,点赞4次
- Python - 读取csv文件和在csv文件写入内容 阅读264次,点赞0次
- Python - 使用flask_sockets库构建websocket服务器 阅读2796次,点赞0次
- Python3 - 导入模块和函数 阅读2898次,点赞0次
- Python3爬虫 - 下载反盗链图片的方式 阅读2643次,点赞1次
- 解决Python爬虫在爬资源过程中使用urlretrieve函数下载文件不完全且避免下载时长过长陷入死循环,并在下载文件的过程中显示下载进度 阅读3760次,点赞0次
- Python - 运算符/ or // or %的含义和区别 阅读1719次,点赞0次
全站随机文章推荐
- Python - argparse命令行参数解析库用法总结 阅读405次,点赞0次
- 资源分享 - Visualizing Quaternions 英文高清PDF下载 阅读1411次,点赞0次
- 资源分享 - Advanced High Dynamic Range Imaging, First Edition 英文高清PDF下载 阅读1223次,点赞0次
- 资源分享 - The HDRI Handbook 2.0- High Dynamic Range Imaging for Photographers and CG Artists高清PDF下载 阅读2110次,点赞0次
- Modern OpenGL - GLSL着色语言1:OpenGL着色器简介 阅读2364次,点赞0次
- OpenCV - 静态图片人脸检测和摄像头人脸检测 阅读2922次,点赞0次
- WordPress - 禁用XML-RPC接口,禁止访问xmlrpc.php,避免DDOS攻击,防止暴力破解 阅读2606次,点赞0次
- yolov5 - 原始仓库检测代码detect.py中文注释 阅读2415次,点赞1次
- C++11/std::thread - 可作为线程函数的几种方式总结 阅读3327次,点赞1次
- 书籍翻译 – Fundamentals of Computer Graphics, Fourth Edition,第4章 Ray Tracing中文翻译 阅读1502次,点赞7次
评论
164