fixed_batch_normalization(x, gamma, beta, mean, var, eps=2e-05, axis=None)¶
Batch normalization function with fixed statistics.
This is a variant of batch normalization, where the mean and variance statistics are given by the caller as fixed variables. This is used on testing mode of the batch normalization layer, where batch statistics cannot be used for prediction consistency.
- x (
cupy.ndarray) – Input variable.
- gamma (
cupy.ndarray) – Scaling parameter of normalized data.
- beta (
cupy.ndarray) – Shifting parameter of scaled normalized data.
- mean (
cupy.ndarray) – Shifting parameter of input.
- var (
cupy.ndarray) – Square of scaling parameter of input.
- eps (float) – Epsilon value for numerical stability.
- axis (int, tuple of int or None) – Axis over which normalization is
performed. When axis is
None, it is determined from input dimensions. For example, if
x.ndim is 4, axis becomes (0, 2, 3) and normalization is performed over 0th, 2nd and 3rd axis of input. If it is 2, axis becomes (0) and normalization is performed over 0th axis of input. When a tuple of int is given to this option, numbers in the tuple must be being sorted in ascending order. For example, (0, 2) is OK, but (2, 0) is not.
- x (