gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-02 22:54:30 29 4
gpt4 key购买 nike

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

但我想知道有没有什么办法,我可以使用列表或数组中的初始值创建 torch 张量?当我在 pytorch 中工作时,还建议我是否有任何 pythonic 方法来实现这一点。

最佳答案

这些是 pytorch 中的常规操作,可在文档中找到。 PyTorch 可以轻松与 numpy 连接。有一种名为 from_numpy 的方法,文档可用 here

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

关于torch - 如何将列表或 numpy 数组转换为一维 torch 张量?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/42894882/

29 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com