as_padded_tensor¶
- class torchrl.envs.llm.transforms.as_padded_tensor(list_of_tensordicts: list[[<class '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
