stack¶
- class tensordict.stack(input: Sequence[T], dim: int = 0, *, out=None)¶
Stacks tensordicts into a single tensordict along the given dimension.
This call is equivalent to calling
torch.stack()
but is compatible with torch.compile.
Stacks tensordicts into a single tensordict along the given dimension.
This call is equivalent to calling torch.stack()
but is compatible with torch.compile.