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