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

How to convert a list or numpy array to a 1d torch tensor?

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

但我想知道有什么办法可以用列表或数组中的初始值创建一个火炬张量?如果我在 pytorch 中工作,是否有任何 pythonic 方法可以实现这一点,也建议我。

这些是 pytorch 中的常规操作,可在文档中找到。 PyTorch 允许与 numpy 轻松连接。有一个方法叫做 from_numpy 并且文档可用 here

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