将一个list的ndarray转为tensor
时间: 2023-05-11 18:04:37 浏览: 101
详解pytorch tensor和ndarray转换相关总结
5星 · 资源好评率100%
可以使用PyTorch中的torch.tensor()函数将一个list的ndarray转为tensor。具体代码如下:
import torch
import numpy as np
# 创建一个list的ndarray
arr = np.array([1, 2, 3, 4, 5])
# 将ndarray转为tensor
tensor = torch.tensor(arr)
print(tensor)
输出结果为:
tensor([1, 2, 3, 4, 5])
阅读全文