API 参考

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