text_plain_output_1.png
import keras

def normalize(x_train, x_test):
    x_train = keras.utils.normalize(x_train)
    x_test = keras.utils.normalize(x_test)
    return (x_train, x_test)

print(x_train_re.shape[1:])
print(x_test_re.shape)
input_shape = x_train.shape[1:]
receptive_field = (3, 3)
pooling_field = (2, 2)