API 参考

• tensordict 包
  • TensorDictBase
  • TensorDict
  • LazyStackedTensorDict
  • PersistentTensorDict
  • TensorDictParams
  • get_defaults_to_none
  • 构造函数和处理器
  • TensorDict 作为上下文管理器
  • 内存映射张量
  • 逐点操作
  • 工具
• tensordict.nn 包
  • TensorDictModuleBase
  • TensorDictModule
  • ProbabilisticTensorDictModule
  • ProbabilisticTensorDictSequential
  • TensorDictSequential
  • TensorDictModuleWrapper
  • CudaGraphModule
  • WrapModule
  • set_composite_lp_aggregate
  • composite_lp_aggregate
  • as_tensordict_module
  • 集成
  • 编译 TensorDictModules
  • 分布
  • 工具
• tensorclass
  • 序列化
  • 边缘情况
  • 自动类型转换