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
声明：本站所有文章，如无特殊说明或标注，均为本站原创发布。任何个人或组织，在未征得本站同意时，禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益，可联系我们进行处理。
文章末尾
评论
72