consolidate_spec¶
- class torchrl.data.consolidate_spec(spec: Composite, recurse_through_entries: bool = True, recurse_through_stack: bool = True)[source]¶
给定一个 TensorSpec,通过添加 0 形状的 spec 来移除独占键。
- 参数:
spec (Composite) – 要合并的 spec。
recurse_through_entries (bool) – 如果为 True,则递归调用函数处理 spec 的所有条目。默认为 True。
recurse_through_stack (bool) – 如果为 True,如果提供的 spec 是延迟加载的,则函数会递归处理其列表中的所有 spec。默认为 True。