我想将一些张量放入列表中,并且我知道如果我想将 nn.Module 类放入列表中,我必须使用 ModuleList
来包装该 list 。
那么,pytorch 中是否有类似“TensorList”的东西,我必须使用它来包装包含张量的列表?
最佳答案
这些张量是什么?这些张量参数是你的 nn.Module
?如果是这样,您需要使用正确的 container .
例如,使用 nn.ParameterList
。这样调用你的模块的 .paramters()
方法也会产生这些张量。否则你会得到类似 this one 的错误.
关于list - pytorch中有类似 'TensorList’的东西吗?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/70594372/