cat

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

沿給定維度將 tensordict 串聯成單個 tensordict。

此呼叫等同於呼叫 torch.cat()，但與 torch.compile 相容。