This also wouldn’t work, as you can concatenate tensors with the same shape other than the specified dimension where the tensors should be concatenated.
E.g. for torch.cat((a, b), dim=x) all dimensions must match besides dim=x.