list - pytorch中有类似 'TensorList’的东西吗?

标签 list pytorch tensor

我想将一些张量放入列表中,并且我知道如果我想将 nn.Module 类放入列表中,我必须使用 ModuleList 来包装该 list 。 那么,pytorch 中是否有类似“TensorList”的东西,我必须使用它来包装包含张量的列表?

最佳答案

这些张量是什么?这些张量参数是你的 nn.Module ?如果是这样,您需要使用正确的 container .
例如,使用 nn.ParameterList 。这样调用你的模块的 .paramters()方法也会产生这些张量。否则你会得到类似 this one 的错误.

关于list - pytorch中有类似 'TensorList’的东西吗?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/70594372/

相关文章:

python - 在 C++ 中索引 tensorflow 输出张量

python - 如何将每个 x 个元素拆分一个列表并将这些 x 个元素添加到一个新列表中?

list - Haskell获取IO列表元素

torch - Pytorch 如何对 3D 张量(变量)中的每个矩阵进行行归一化?

python - 使用 torchtext 和文本序列作为输入

regression - Pytorch loss inf nan

python - 替换列表中子列表中所有出现的项目

c++ - 如何将列表中的所有元素移动到数组或 vector 或其他任何内容

python - 迭代 torch 张量

python-3.x - TensorFlow训练时进度条前的数字是什么意思