tf.nn.moments

Calculates the mean and variance of x.

tf.nn.moments(
    x, axes, shift=None, keepdims=False, name=None
)

Used in the notebooks

Used in the tutorials
Distributed training with DTensors

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).

Args
`x`	A `Tensor`.
`axes`	Array of ints. Axes along which to compute mean and variance.
`shift`	Not used in the current implementation.
`keepdims`	produce moments with the same dimensionality as the input.
`name`	Name used to scope the operations that compute the moments.

Returns
Two `Tensor` objects: `mean` and `variance`.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-04-26 UTC.

tf.nn.momentsStay organized with collections Save and categorize content based on your preferences.

Used in the notebooks

Args

Returns

tf.nn.moments