reference/generated/torchrl.envs.llm.transforms.as_nested_tensor

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

as_nested_tensor

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

Stacks a list of tensordicts into a single tensordict with nested tensors.

Parameters:

list_of_tensordicts (list[TensorDictBase]) – A list of tensordicts to stack.

Returns:

A tensordict with nested tensors.

Return type:

TensorDictBase