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/api_docs/python/tf/compat/v1/metrics/mean_tensor
