将一个list的ndarray转为tensor
时间: 2023-05-11 21:04:37 浏览: 104
在pytorch中为Module和Tensor指定GPU的例子
可以使用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])
阅读全文