Hi,
Could you please use triple backticks before and after code to make it more readable!

But I couldn’t load my data appropriately.

What do you mean? The reshape method you show at the beginning should work if your input is indeed of size [1, 8, 1, 3, 224, 224].