Spec-Zone.ru › TensorFlow

tf.nn.moments

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

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

Используется в ноутбуках

Используется в учебниках
  • Распределенное обучение с DTensors

Среднее значение и дисперсия вычисляются путем агрегации содержимого x по axes. Если x имеет размерность 1 и 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.

© 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/nn/moments

Spec-Zone.ru

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