cat

class tensordict.cat(input, dim: int = 0, *, out=None)

将tensordicts沿给定维度连接成一个tensordict。

此调用等效于调用 torch.cat()，但与 torch.compile 兼容。