Correct Input dimension of Images for Training?

I need to know what is the correct input dimension of Input images some discussion forum tell me its channelHeightWidth other telling me HeightWidthchannel

Which one do I need to chose while doing training of Model and for Visualization.

PyTorch expects inputs as [batch_size, channels, height, width] for the “usual” layers in CNNs.

If you want to visualize numpy arrays via e.g. matplotlib, then you should remove the batch dimension and permute the array to a channels-last format.
However, this is matplotlib specific and other visualization libraries might use other formats.