tf.keras.ops.moments

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

Просмотр псевдонимов

Основные псевдонимы

tf.keras.ops.nn.moments

tf.keras.ops.moments(
    x, axes, keepdims=False, synchronized=False
)

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

Аргументы
`x`	Входной тензор.
`axes`	Список осей, по которым вычисляется среднее и дисперсия.
`keepdims`	Если это установлено в `True`, оси, по которым производилось сокращение, остаются в результате в виде измерений с размером один.
`synchronized`	Применимо только с бэкэндом TensorFlow. Если `True`, синхронизирует глобальную статистику по партиям (среднее и дисперсия) по всем устройствам на каждой итерации обучения в стратегии распределенного обучения. Если `False`, каждая реплика использует свою локальную статистику по партиям.

Возвращаемые значения
Кортеж, содержащий два тензора - среднее и дисперсию.

Пример:

x = keras.ops.convert_to_tensor([0, 1, 2, 3, 100], dtype="float32")
keras.ops.moments(x, axes=[0])
(array(21.2, dtype=float32), array(1553.3601, dtype=float32))

© 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/keras/ops/moments