fix BatchNormalization layer axis normalization
fix BatchNormalization layer axis normalization

qinwf-nuan/keras-js,transcranial/keras-js,qinwf-nuan/keras-js,qinwf-nuan/keras-js,transcranial/keras-js,transcranial/keras-js