快捷方式

maybe_dense_stack

class tensordict.maybe_dense_stack(input, dim: int = 0, *, out=None, **kwargs)

尝试使 TensorDicts 密集堆叠,并在需要时回退到懒惰堆叠。

有关详细信息,请参阅 maybe_dense_stack()

文档

访问全面的 PyTorch 开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源