我有一个浮点值列表（或一个numpy数组）。我想创建一个包含所有这些值的1d火炬张量。我可以创建火炬张量并运行循环来存储值。
但我想知道有什么办法，我可以用列表或数组中的初始值创建一个火炬张量？还建议我，如果有任何pythonic方式来实现这一点，因为我在pytorch工作。
答案 0 :(得分：15)
这些是pytorch中的常规操作，可在文档中找到。 PyTorch可以轻松连接numpy。有一种名为from_numpy的方法，文档可用here
import numpy as np
import torch
array = np.arange(1, 11)
tensor = torch.from_numpy(array)