stack

class tensordict.stack(input: Sequence['T'], dim: int = 0, *, out=None)

Stacks tensordicts into a single tensordict along the given dimension.