cat

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

沿给定维度将 tensordict 串联成单个 tensordict。

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