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。