tf.compat.v1.nn.moments

Вычисляет среднее значение и дисперсию x.

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

Среднее значение и дисперсия вычисляются путем агрегирования содержимого x вдоль axes. Если x — одномерный массив и axes = [0], то это просто среднее значение и дисперсия вектора.

Примечание: параметр shift в настоящее время не используется; вычисляется и используется истинное среднее значение.

При использовании этих моментов для пакетной нормализации (см. tf.nn.batch_normalization):

для так называемой «глобальной нормализации», используемой со сверточными фильтрами формы [batch, height, width, depth], передайте axes=[0, 1, 2].
для простой пакетной нормализации передайте axes=[0] (только пакет).

Аргументы
`x`	A `Tensor`.
`axes`	Массив целых чисел. Оси, вдоль которых вычисляются среднее значение и дисперсия.
`shift`	Не используется в текущей реализации
`name`	Имя, используемое для задания области видимости операций, вычисляющих моменты.
`keep_dims`	создавать моменты с той же размерностью, что и входные данные.
`keepdims`	Псевдоним для keep_dims.

Возвращает
Два объекта `Tensor`: `mean` и `variance`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/compat/v1/nn/moments