API 參考

• tensordict 封裝
  • TensorDictBase
  • TensorDict
  • LazyStackedTensorDict
  • PersistentTensorDict
  • TensorDictParams
  • get_defaults_to_none
  • 建構子和處理器
  • TensorDict 作為上下文管理器
  • 記憶體映射張量
  • 工具
• tensordict.nn 封裝
  • TensorDictModuleBase
  • TensorDictModule
  • ProbabilisticTensorDictModule
  • TensorDictSequential
  • TensorDictModuleWrapper
  • CudaGraphModule
  • Ensembles
  • 編譯 TensorDictModules
  • 分布
  • 工具
• tensorclass
  • 序列化
  • 邊緣案例
  • 自動轉換