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
