consolidate_spec¶
- class torchrl.data.consolidate_spec(spec: CompositeSpec, recurse_through_entries: bool = True, recurse_through_stack: bool = True)[source]¶
给定一个 TensorSpec,通过添加 0 形状的规格来删除独占键。
- 参数::
spec (CompositeSpec) – 要合并的规格。
recurse_through_entries (bool) – 如果为 True,则递归调用该函数对规格的所有条目进行操作。默认值为 True。
recurse_through_stack (bool) – 如果为 True,如果提供的规格是延迟的,则递归调用该函数对列表中的所有规格进行操作。默认值为 True。