My network predicts the values of missing pixels of a grayscale image. However, since the image is normalized, the network gives the normalized values between 0 and 1 with float32 datatype. I guess if the network could output discrete values in [0,1,2,…,255], it would be much more simpler and faster. Should I just put the threshold function that discretizes the float32 output? Do you have any ideas?