View source on GitHub |
Computes the element-wise (weighted) mean of the given tensors.
tf.compat.v1.metrics.mean_tensor(
values,
weights=None,
metrics_collections=None,
updates_collections=None,
name=None
)
In contrast to the mean
function which returns a scalar with the
mean, this function returns an average tensor with the same shape as the
input tensors.
The mean_tensor
function creates two local variables,
total_tensor
and count_tensor
that are used to compute the average of
values
. This average is ultimately returned as mean
which is an idempotent
operation that simply divides total
by count
.
For estimation of the metric over a stream of data, the function creates an
update_op
operation that updates these variables and returns the mean
.
update_op
increments total
with the reduced sum of the product of values
and weights
, and it increments count
with the reduced sum of weights
.
If weights
is None
, weights default to 1. Use weights of 0 to mask values.