【问题标题】:How to convert a list or numpy array to a 1d torch tensor?如何将列表或 numpy 数组转换为一维火炬张量?
【发布时间】:2017-08-11 05:11:46
【问题描述】:

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

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

【问题讨论】:

    标签: torch pytorch


    【解决方案1】:

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

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

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 2021-11-30
      • 2022-07-20
      • 1970-01-01
      • 2020-12-24
      • 2021-12-12
      • 2020-12-31
      • 2021-02-02
      • 2022-01-20
      相关资源
      最近更新 更多