BatchNormalization¶
Carries out batch normalization as described in the paper https://arxiv.org/abs/1502.03167. Depending on the mode it is being run, There are five required inputs ‘X’, ‘scale’, ‘B’, ‘input_mean’ and ‘input_var’. Note that ‘input_mean’ and ‘input_var’ are expected to be the estimated statistics in inf…
Abstract Signature:
BatchNormalization(X: Tensor, scale, B, input_mean, input_var, epsilon: float, momentum: float, training_mode: int)