@ptrblck thank you for this my_print func - I will try this as soon as the gpus become free. What I see is that for some reason, Ihave an extra pixel across, so I seem to get [3.385,385] instead of [3,384,385]. Currently, I just centercrop it to 384 and resolve the issue, but yes something does not add up!
Will get back with results.
Thank you again. Your help is invaluable.