Applies batch normalization on x given mean, var, beta and gamma.
output <- (x - mean) / (sqrt(var) + epsilon) * gamma + beta
k_batch_normalization(x, mean, var, beta, gamma, axis = -1, epsilon = 0.001)
Input tensor or variable.
Mean of batch.
Variance of batch.
Tensor with which to center the input.
Tensor by which to scale the input.
Axis (axis indexes are 1-based). Pass -1 (the default) to select the last axis.
This function is part of a set of Keras backend functions that enable lower level access to the core operations of the backend tensor engine (e.g. TensorFlow, CNTK, Theano, etc.).
You can see a list of all available backend functions here: https://keras.rstudio.com/articles/backend.html#backend-functions.