def flatten_image(x, label=True): if label: return (tf.divide(tf.dtypes.cast(tf.reshape(x["image"], (1,28*28)), tf.float32), 256.0) , x["label"]) else: ...
確定! 回上一頁