Spec-Zone.ru › TensorFlow 1.15

tf.compat.v2.nn.moments

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

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

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

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

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

  • для так называемой «глобальной нормализации», используемой с свёрточными фильтрами с формой [batch, height, width, depth], передайте axes=[0, 1, 2].
  • для простой пакетной нормализации передайте axes=[0] (только пакет).
Аргументы
x Tensor
axes Массив целых чисел. Оси, по которым вычисляется среднее и дисперсия.
shift Не используется в текущей реализации.
keepdims Производит моменты той же размерности, что и входные данные.
name Имя, используемое для определения областей операций, вычисляющих моменты.
Возвращаемое значение
Два объекта Tensor: mean и variance.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/compat/v2/nn/moments

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API