tf.metrics.mean_tensor

Вычисляет поэлементное (взвешенное) среднее значение заданных тензоров.

tf.metrics.mean_tensor(
    values, weights=None, metrics_collections=None, updates_collections=None,
    name=None
)

В отличие от функции mean, которая возвращает скаляр со средним значением, эта функция возвращает тензор среднего значения с той же формой, что и входные тензоры.

Функция mean_tensor создаёт две локальные переменные, total_tensor и count_tensor, которые используются для вычисления среднего значения values. Это среднее значение в конечном итоге возвращается как mean, что является идемпотентной операцией, которая просто делит total на count.

Для оценки метрики по потоку данных функция создаёт операцию update_op, которая обновляет эти переменные и возвращает mean. update_op увеличивает total на уменьшенную сумму произведения values и weights, и увеличивает count на уменьшенную сумму weights.

Если weights равно None, веса по умолчанию равны 1. Используйте веса 0 для маскирования значений.