So I am trying to normalize a tensor the needed some extra dimensions. So, I used unsqueeze as you’ll observe in the following code in order to add this extra dimension, which corresponds to the channels of the image.