consolidate_spec¶
- class torchrl.data.consolidate_spec(spec: Composite, recurse_through_entries: bool = True, recurse_through_stack: bool = True)[source]¶
给定一个 TensorSpec，通过添加 0 形状的规范来删除独占键。
- 参数:
spec (Composite) – 要整合的规范。
recurse_through_entries (bool) – 如果为 True，则对规范的所有条目递归调用该函数。默认为 True。
recurse_through_stack (bool) – 如果为 True，如果提供的规范是惰性的，则在其列表中的所有规范上递归调用该函数。默认为 True。
