|TensorFlow 1 version||View source on GitHub|
Calculates the mean and variance of
tf.nn.moments( x, axes, shift=None, keepdims=False, name=None )
The mean and variance are calculated by aggregating the contents of
x is 1-D and
axes =  this is just the mean
and variance of a vector.
When using these moments for batch normalization (see
- for so-called "global normalization", used with convolutional filters with
[batch, height, width, depth], pass
axes=[0, 1, 2].
- for simple batch normalization pass
||Array of ints. Axes along which to compute mean and variance.|
||Not used in the current implementation.|
||produce moments with the same dimensionality as the input.|
||Name used to scope the operations that compute the moments.|