InvertibleGrayscale_Reimplementation_in_Pytorch