如何将列表或numpy数组转换为一维火炬张量？

Question

我有一个浮点值列表（或一个numpy数组）。我想创建一个包含所有这些值的一维火炬张量。我可以创建火炬张量并运行循环来存储值。

但我想知道有没有什么办法，我可以用列表或数组中的初始值创建一个火炬张量？还建议我是否有任何 pythonic 方法可以实现这一点，因为我正在 pytorch 中工作。

Answer 1

这些是 pytorch 中的常规操作，可在文档中找到。 PyTorch 可以轻松与 numpy 连接。有一种名为

from_numpy

的方法，并且可以在 here

获取文档

import numpy as np 
import torch 
array = np.arange(1, 11)
tensor = torch.from_numpy(array)