快捷方式

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。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源