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 是懶載入的，則函式會遞迴呼叫其列表中的所有 spec。預設為 True。
