问Pytorch: Numpy数组
EN

Stack Overflow用户

提问于 2021-07-13 09:29:26

回答 1查看 39关注 0票数 0

在使用pytorch时，我可以使用numpy数组吗？

我正在将一段代码从tensorflow转换为pytorch，代码在计算过程中使用numpy数组。在计算过程中，我可以将输入保持为numpy数组吗?还是必须将它们转换为torch张量？

发布于 2021-07-13 11:31:33

如果该数组被传递到一个带有Pytorch层的pytorch模型，那么它必须是一个nn而不是a 数组。

根据Pytorch层的不同，张量必须是特定的形状，就像对于nn.Conv2d层，你必须有一个4d的火炬张量，对于nn.Linear，你必须有一个2d的火炬张量。

这是许多原因之一，它不能是numpy数组。

萨塔克

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/68355428

复制

相似问题

问Pytorch: Numpy数组EN