Rate this Page
★★★★★

ModuleList#

classtorch.nn.ModuleList(modules=None)[source]#

Holds submodules in a list.

ModuleList can be indexed like a regular Python list, butmodules it contains are properly registered, and will be visible by allModule methods.

Parameters

modules (iterable,optional) – an iterable of modules to add

Example:

classMyModule(nn.Module):def__init__(self)->None:super().__init__()self.linears=nn.ModuleList([nn.Linear(10,10)foriinrange(10)])defforward(self,x):# ModuleList can act as an iterable, or be indexed using intsfori,linenumerate(self.linears):x=self.linears[i//2](x)+l(x)returnx
append(module)[source]#

Append a given module to the end of the list.

Parameters

module (nn.Module) – module to append

Return type

Self

extend(modules)[source]#

Append modules from a Python iterable to the end of the list.

Parameters

modules (iterable) – iterable of modules to append

Return type

Self

insert(index,module)[source]#

Insert a given module before a given index in the list.

Parameters

  • index (int) – index to insert.

  • module (nn.Module) – module to insert