Yes, cat works fine MariosOreo. Have I found a bug using the stack method as per above?
I am thinking if things like this happens maybe I should better use np.stack()  to prepare the data instead torch.stack(). I guess it is still early days…
Any feedback?