¿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
TensorList
en 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.Module
u objetos personalizados que contienen tensores como atributos, puedes utilizar las clasesModuleList
oModuleDict
proporcionadas por PyTorch para asegurar un manejo adecuado de los objetos contenidos.