Suppose I have a tensor a with shape (2, 9, 4), is there an efficient way to apply torch.cat to all a[i]s like this:
torch.cat([a[0], a[1]], dim=1)

so it became a tensor with shape (9, 8)
Thanks in advance.