chainer.functions.group_normalization

chainer.functions.group_normalization(x, groups, gamma, beta, eps=1e-05)[source]

Group normalization function.

This function implements a “group normalization” which divides the channels into groups and computes within each group the mean and variance, then normalize by these statistics, scales and shifts them.

Parameters
  • x (Variable or N-dimensional array) – Batch tensors. First dimension of this value must be the size of minibatch and second dimension must be the number of channels. Moreover, this value must have one or more following dimensions, such as height and width.

  • groups (int) – The number of channel groups. This value must be a divisor of the number of channels.

  • gamma (Variable or N-dimensional array) – Scaling parameter.

  • beta (Variable or N-dimensional array) – Shifting parameter.

  • eps (float) – Epsilon value for numerical stability of normalization.

Returns

The output variable which has the same shape as \(x\).

Return type

Variable

See: Group Normalization

See also

GroupNormalization to manage the model parameters gamma and beta.