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 | Tensor произвольной размерности. |
weights | Необязательный Tensor, ранг которого равен 0 или совпадает с рангом values и который должен допускать широковещательное распространение до values (то есть все размерности должны быть либо 1, либо совпадать с соответствующей размерностью values). |
metrics_collections | Необязательный список коллекций, в которые следует добавить mean. |
updates_collections | Необязательный список коллекций, в которые следует добавить update_op. |
name | Необязательное имя variable_scope. |
| Возвращает | |
|---|---|
mean | Число с плавающей запятой Tensor, представляющее текущее среднее значение: значение total, деленное на count. |
update_op | Операция, которая соответствующим образом увеличивает переменные total и count и значение которой совпадает с mean_value. |
| Вызывает исключения | |
|---|---|
ValueError | Если weights не равно None и его форма не совпадает с values либо если metrics_collections или updates_collections не являются списком или кортежем. |
RuntimeError | Если включен eager execution. |
© 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
