BTNug:

(a) you do concatenation cat1 of x1 and x2 at axis 2 , the output is 1x56x28x3 (incorrect)


Isn’t the cat of x1 and x2 at axis 2 --> 1x3x56x28 unless there’s some kind of transpose happening which I don’t see?
By the way, welcome to the community .