consolidate_spec¶
- class torchrl.data.consolidate_spec(spec: Composite, recurse_through_entries: bool = True, recurse_through_stack: bool = True)[原始碼]¶
給定 TensorSpec,透過新增 0 形狀的規格來移除獨佔鍵。
- 參數:
spec (Composite) – 要合併的規格。
recurse_through_entries (bool) – 如果為 True,則在規格的所有條目上遞迴呼叫此函數。預設值為 True。
recurse_through_stack (bool) – 如果為 True,如果提供的規格是惰性的,則在清單中的所有規格上遞迴呼叫此函數。預設值為 True。