I used keras from tensorflow to establishing the CNN, now I add the layers as following:
model.add(layers.Conv2D(6, (5, 5), activation=\'relu\', input_shape=(200,