Package Reference
torchrl.nn.
ModuleExtended
Bases: torch.nn.modules.module.Module
torch.nn.modules.module.Module
A torch module with added functionalities.
SequentialExtended
Bases: torchrl.nn.container.ModuleExtended
torchrl.nn.container.ModuleExtended
A torch sequential module with added functionalities.
FlattenLinear
Bases: torch.nn.modules.linear.Linear
torch.nn.modules.linear.Linear
Flatten the input and apply a linear layer.
ActionLinear
A linear layer that automatically calculates the output shape based on the action_info.