Hi Mohammad!



 m.hassanin:

A=[2,4,5,5], How it becomes A=[2,20,5]?


I’m not sure what you mean by “concatenate.”  I’m guessing that the
values in brackets ([]) indicate the shapes of the tensors.  Does:
A.reshape ([2, 20, 5])

give you what you are looking for?
If not, could you give an explicit numerical example of what your
result tensor should be?
Best.
K. Frank