快捷方式

as_nested_tensor

class torchrl.envs.llm.transforms.as_nested_tensor(list_of_tensordicts: list[tensordict.base.TensorDictBase])[源]

将 tensordicts 列表堆叠成一个具有嵌套张量的 tensordict。

参数:

list_of_tensordicts (list[TensorDictBase]) – 要堆叠的 tensordicts 列表。

返回:

具有嵌套张量的 tensordict。

返回类型:

TensorDictBase

文档

访问全面的 PyTorch 开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源