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。