tf.compat.v1.nn.moments

Calculate the mean and variance of x.

tf.compat.v1.nn.moments(
    x, axes, shift=None, name=None, keep_dims=None, keepdims=None
)

The mean and variance are calculated by aggregating the contents of x across axes. If x is 1-D and axes = [0] this is just the mean and variance of a vector.

When using these moments for batch normalization (see tf.nn.batch_normalization):

  • for so-called "global normalization", used with convolutional filters with shape [batch, height, width, depth], pass axes=[0, 1, 2].
  • for simple batch normalization pass axes=[0] (batch only).

xA Tensor.
axesArray of ints. Axes along which to compute mean and variance.
shiftNot used in the current implementation
nameName used to scope the operations that compute the moments.
keep_dimsproduce moments with the same dimensionality as the input.
keepdimsAlias to keep_dims.

Two Tensor objects: mean and variance.