快捷方式

consolidate_spec

class torchrl.data.consolidate_spec(spec: Composite, recurse_through_entries: bool = True, recurse_through_stack: bool = True)[原始碼]

給定一個 TensorSpec,透過新增形狀為 0 的 spec 來移除專屬鍵。

引數:
  • spec (Composite) – 要整合的 spec。

  • recurse_through_entries (bool) – 如果為 True,則在 spec 的所有條目上遞迴呼叫此函式。預設為 True。

  • recurse_through_stack (bool) – 如果為 True,並且提供的 spec 是 lazy 的,則在此列表中的所有 spec 上遞迴呼叫此函式。預設為 True。

文件

查閱 PyTorch 的全面開發者文件

檢視文件

教程

獲取針對初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源