如何将列表或numpy数组转换为1维torch张量？

Question

我有一个浮点数值的列表（或numpy数组），我想创建一个1D torch张量，其中包含所有这些值。 我可以创建torch张量并运行循环以存储这些值。

但我想知道是否有任何方法可以从列表或数组中创建具有初始值的torch张量？ 如果在pytorch中工作，还请为我提供pythonic方式。

- Wasi Ahmad

Answer 1

这些是PyTorch中的常规操作，文档中提供了相应的信息。PyTorch可以轻松地与numpy进行接口交互。有一个名为from_numpy的方法，文档位于这里

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

- Kashyap