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 | A Tensor произвольных размерностей. |
weights | Необязательный Tensor, ранг которого равен 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. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/compat/v1/metrics/mean_tensor
