快捷方式

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。

文档

访问全面的 PyTorch 开发者文档

查看文档

教程

为初学者和高级开发者提供深入的教程

查看教程

资源

查找开发资源并让您的问题得到解答

查看资源