tf.compat.v1.metrics.mean_tensor
Вычисляет поэлементное (взвешенное) среднее значение заданных тензоров.
tf.compat.v1.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 для маскирования значений.
| Аргументы | |
|---|---|
values | Массив произвольных размерностей. |
weights | Необязательные веса, ранг которых равен 0 или совпадает с рангом values, и которые должны быть совместимы по ширине с values (то есть все измерения должны быть либо 1, либо совпадать с соответствующим измерением values). |
metrics_collections | Необязательный список коллекций, в которые следует добавить mean. |
updates_collections | Необязательный список коллекций, в которые следует добавить update_op. |
name | Необязательное имя области переменных. |
| Возвращаемое значение | |
|---|---|
mean | Вещественный тензор Tensor, представляющий текущее среднее значение, значение total , поделенное на count. |
update_op | Операция, которая соответствующим образом увеличивает переменные total и count и значение которой соответствует mean_value. |
| Исключения | |
|---|---|
ValueError | Если weights не None и его форма не соответствует values, или если metrics_collections или updates_collections не являются списком или кортежем. |
RuntimeError | Если включена жадная (eager) вычислительная модель. |
© 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/versions/r2.9/api_docs/python/tf/compat/v1/metrics/mean_tensor
