nn

Module for PyTorch neural network utilities.

Classes:

Name	Description
ModuleDefaultDict	Holds submodules in a dictionary and calls a factory function to supply missing values.

ModuleDefaultDict

Bases: ModuleDict, Generic[ModuleT]

Holds submodules in a dictionary and calls a factory function to supply missing values.

Parameters:

Name	Type	Description	Default
default_factory	Callable[[], ModuleT] | None	A no argument constructor for missing Module items. If None, missing items will raise a KeyError.	None
**kwargs	ModuleT	Initial dictionary values.	{}

Methods:

Name	Description
__getitem__	Retrieve a module by key, creating it if missing.

__getitem__

__getitem__(key: str) -> ModuleT

Retrieve a module by key, creating it if missing.

Parameters:

Name	Type	Description	Default
key	str	The name of the module to retrieve.	required

Returns:

Type	Description
ModuleT	The requested module.