cat

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

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

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