tf.keras.ops.average

Вычислить взвешенное среднее по указанной оси.

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

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

tf.keras.ops.numpy.average

tf.keras.ops.average(
    x, axis=None, weights=None
)

Аргументы
`x`	Входной тензор.
`axis`	Целое число вдоль которого вычисляется среднее значение `x`. По умолчанию, `axis=None`, вычисляется среднее по всем элементам входного тензора. Если ось отрицательная, она отсчитывается от последней до первой оси.
`weights`	Тензор весов, связанных со значениями в `x`. Каждое значение в `x` вносит вклад в среднее значение в соответствии со своим весом. Массив весов может быть одномерным (в этом случае его длина должна быть размером вдоль заданной оси) или иметь такую же форму, как и `x`. Если значение `weights=None` (по умолчанию), то предполагается, что все данные в `x` имеют вес, равный единице. Вычисление для одномерного случая: `avg = sum(a * weights) / sum(weights)`. Единственное ограничение для весов заключается в том, что `sum(weights)` не должно быть равным 0.

Возвращаемое значение
Возвращает среднее значение вдоль указанной оси.

Примеры:

data = keras.ops.arange(1, 5)
data
array([1, 2, 3, 4], dtype=int32)
keras.ops.average(data)
array(2.5, dtype=float32)
keras.ops.average(
    keras.ops.arange(1, 11),
    weights=keras.ops.arange(10, 0, -1)
)
array(4., dtype=float32)

data = keras.ops.arange(6).reshape((3, 2))
data
array([[0, 1],
       [2, 3],
       [4, 5]], dtype=int32)
keras.ops.average(
    data,
    axis=1,
    weights=keras.ops.array([1./4, 3./4])
)
array([0.75, 2.75, 4.75], dtype=float32)
keras.ops.average(
    data,
    weights=keras.ops.array([1./4, 3./4])
)
Traceback (most recent call last):

ValueError: Axis must be specified when shapes of a and weights differ.

© 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/average