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
)
如果帮到了你,就请收藏评论加点赞,多谢支持。
转载请注明原文地址:https://tech.qufami.com/read-6173.html