tf.keras.layers.MaxPooling2D with padding='same'

This thread might be helpful.