model.tools.generator

Model generation tools

Custom types

List for storing the state of the model

ListState: List[torch.Tensor | None | ListState]

Classes

LayerGen	Base class for model layer generators
ModelGenerator	Tool for generating and processing a model from a list of layer generators
ResidualModule	The class combines data stored in the storage by summing or concatenating
StateStorage	Class wrapper for neurons with a state of Norse.
Storage	Stores the forward pass values
StorageGetter	Returns the specified stored tensor