as_padded_tensor¶
- class torchrl.envs.llm.transforms.as_padded_tensor(list_of_tensordicts: list[tensordict.base.TensorDictBase], dim=0, stack_dim: int = 0)[源代码]¶
将 tensordicts 列表堆叠成一个具有填充张量的单个 tensordict。
- 参数:
list_of_tensordicts (list[[TensorDictBase]]) – 要堆叠的 tensordicts 列表。
dim (int, optional) – 要填充的维度。默认为 0。
stack_dim (int, optional) – 要堆叠的维度。默认为 0。
- 返回:
一个具有填充张量的 tensordict。
- 返回类型:
TensorDictBase
