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

Question

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

但我想知道有什么办法，我可以从列表或数组中创建一个带有初始值的火炬张量吗？当我在 pytorch 工作时，如果有任何 pythonic 方式来实现这一点，也建议我。

Answer 1

这些是 pytorch 中的一般操作，可在文档中找到。PyTorch 允许与 numpy 轻松交互。有一个方法被调用from_numpy，文档可在此处获得

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