Pytorch normalize dimension. Min-Max Normalization.