ValueError: Shape must be rank 1 but is rank 4
I'm using Keras 2.2.4 and just a simple code for BatchNormalization throws this error
ValueError: Shape must be rank 1 but is rank 4 for 'batch_normalization_1/cond/FusedBatchNorm' (op: 'FusedBatchNorm') with input shapes
The back-end is Tensorflow. Here is the code I'm running
from keras.layers import BatchNormalization, Input
x = Input(shape=(1, 32, 32))
BatchNormalization(axis=1)(x)
What's wrong with this?