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

Question

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

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

Answer 1

这些是pytorch中的常规操作,可在文档中找到.PyTorch可以轻松连接numpy.有一个方法被调用from_numpy,文档可以在这里找到

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