reference/modules_utils

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

Utilities and Helpers

Utility modules and helper functions for building RL networks.

ActorValueOperator(*args, **kwargs)	Actor-value operator.
ActorCriticOperator(*args, **kwargs)	Actor-critic operator.
ActorCriticWrapper(*args, **kwargs)	Actor-value operator without common module.
get_primers_from_module(module[, warn, strict])	Get all tensordict primers from all submodules of a module.
get_env_transforms_from_module(module[, ...])	Return all TransformedEnv transforms needed for a recurrent module.

get_recurrent_matmul_precision()	Resolve the currently effective precision to a concrete mode.
set_recurrent_matmul_precision(mode)	Set the process-global precision for the triton RNN backend.

torchrl.modules.RecurrentMatmulPrecision

alias of Literal[‘ieee’, ‘tf32’, ‘tf32x3’]

torchrl.modules.RecurrentMatmulPrecisionUserMode

alias of Literal[‘auto’, ‘fast’, ‘high-prec’, ‘ieee’, ‘tf32’, ‘tf32x3’]

SquashDims([ndims_in])

A squashing layer.