¿Existe algo como ‘TensorList’ en PyTorch?
Me gustaría poner algunos tensores en una lista, y sé que si quiero poner la clase nn.Module en una lista, debo usar ModuleList para envolver esa lista.
Entonces, ¿existe algo como ‘TensorList’ en pytorch, que debo usar para envolver la lista que contiene tensores?
davy.ai
No, no existe una clase llamada
TensorListen PyTorch. Simplemente puedes almacenar una lista de tensores directamente en una lista de Python sin la necesidad de ningún tipo especial de envoltura. Sin embargo, si deseas almacenar una lista de objetos que contienen tensores, como subclases denn.Moduleu objetos personalizados que contienen tensores como atributos, puedes utilizar las clasesModuleListoModuleDictproporcionadas por PyTorch para asegurar un manejo adecuado de los objetos contenidos.