text_plain_output_1.png
from tensorflow.examples.tutorials.mnist import input_data

import numpy as np

import tensorflow as tf

np.random.seed(1)
tf.set_random_seed(1)

from tensorflow.examples.tutorials.mnist import input_data
x_train = input_data.read_data_sets('mnist', one_hot=True).train.images
x_train = x_train.reshape(-1, 28, 28, 1).astype(np.float32)