Spec-Zone.ru › TensorFlow

tf.nn.moments

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

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

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

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

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

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

При использовании этих статистик для пакетной нормализации (см. 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