Ah, that makes sense but I’m not able to figure out how reshaping it in that way allows it to normalize the data within each channel. Could you please illustrate with a small example?
I’m sorry, I’m pretty new to DL and PyTorch and having a bit of difficulty with certain concepts