mnist.py 文件源码

def parse(mode, image, label):
    """Parse input record to features and labels."""
    image = tf.to_float(image)
    label = tf.to_int64(label)
    image = tf.image.per_image_standardization(image)
    return {"image": image}, {"label": label}