TensorFlow函數(shù)：tf.metrics.mean_tensor

tf.metrics.mean_tensor函數(shù)

tf.metrics.mean_tensor(
    values,
    weights=None,
    metrics_collections=None,
    updates_collections=None,
    name=None
)

定義在：tensorflow/python/ops/metrics_impl.py.

計(jì)算給定張量的element-wise(加權(quán))平均值.

與返回帶有均值的標(biāo)量的mean函數(shù)相比,此函數(shù)返回與輸入張量具有相同形狀的平均張量.

該mean_tensor函數(shù)創(chuàng)建兩個(gè)局部變量,total_tensor和count_tensor,它們被用于計(jì)算values的平均值.最終返回這個(gè)平均值作為mean,一種冪等運(yùn)算,簡(jiǎn)單地用count除total.

為了估計(jì)數(shù)據(jù)流上的度量,該函數(shù)創(chuàng)建一個(gè)update_op操作來(lái)更新這些變量并返回mean.然后update_op通過(guò)values和weights乘積的減少總和來(lái)遞增total,并且它通過(guò)weights的減少總和來(lái)遞增count.

如果weights是None,則權(quán)重默認(rèn)為1,使用權(quán)重0來(lái)屏蔽值.

參數(shù)：

• values：任意維度的Tensor.
• weights：可選的Tensor,其秩為0或與values相同的秩,并且必須可廣播到values(即,所有維度必須為1或者與相應(yīng)的values維度相同).
• metrics_collections：mean應(yīng)添加到的集合的可選列表.
• updates_collections：update_op應(yīng)添加到的集合的可選列表.
• name：可選的variable_scope名稱(chēng).

返回：

• mean：表示當(dāng)前均值的浮點(diǎn)Tensor,total除以count的值.
• update_op：適當(dāng)增加total和count變量,并且其值與mean_value匹配的操作.

可能引發(fā)的異常：

• ValueError：如果weights不是None,并且它的形狀與values不匹配,或者如果metrics_collections或者updates_collections中任意一個(gè)不是一個(gè)列表或元組.
• RuntimeError：如果啟用了急切執(zhí)行.