I am not sure but I think I might do flatten the array and do padding then do concatenation. I really don’t know how should I concatenate 2dim tensor vertically. Do you have any idea or coding attempt that might do some trick on that?