API Reference

• tensordict package
  • TensorDictBase
  • TensorDict
  • LazyStackedTensorDict
  • PersistentTensorDict
  • TensorDictParams
  • get_defaults_to_none
  • Constructors and handlers
  • TensorDict as a context manager
  • Memory-mapped tensors
  • Pointwise Operations
  • Utils
• tensordict.nn package
  • TensorDictModuleBase
  • TensorDictModule
  • ProbabilisticTensorDictModule
  • ProbabilisticTensorDictSequential
  • TensorDictSequential
  • TensorDictModuleWrapper
  • CudaGraphModule
  • WrapModule
  • set_composite_lp_aggregate
  • composite_lp_aggregate
  • as_tensordict_module
  • Ensembles
  • Compiling TensorDictModules
  • Distributions
  • Utils
• tensorclass
  • Serialization
  • Edge cases
  • Auto-casting