Python tf.keras.metrics.MeanTensor用法及代碼示例

計算給定張量的元素(加權)平均值。

繼承自：Metric、Layer、Module

用法

tf.keras.metrics.MeanTensor(
    name='mean_tensor', dtype=None, shape=None
)

參數

name (可選)指標實例的字符串名稱。
dtype (可選)度量結果的數據類型。
shape (可選)整數列表、整數元組或 int32 類型的一維張量。如果未指定，則從第一次調用update_state 時的值推斷形狀。

屬性

count
total

MeanTensor 返回一個與輸入張量形狀相同的張量。通過保留局部變量 total 和 count 來更新平均值。 total 跟蹤加權值的總和，count 存儲加權計數的總和。

單機使用：

m = tf.keras.metrics.MeanTensor()
m.update_state([0, 1, 2, 3])
m.update_state([4, 5, 6, 7])
m.result().numpy()
array([2., 3., 4., 5.], dtype=float32)

m.update_state([12, 10, 8, 6], sample_weight= [0, 0.2, 0.5, 1])
m.result().numpy()
array([2.       , 3.6363635, 4.8      , 5.3333335], dtype=float32)

m = tf.keras.metrics.MeanTensor(dtype=tf.float64, shape=(1, 4))
m.result().numpy()
array([[0., 0., 0., 0.]])
m.update_state([[0, 1, 2, 3]])
m.update_state([[4, 5, 6, 7]])
m.result().numpy()
array([[2., 3., 4., 5.]])

相關用法

注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.keras.metrics.MeanTensor。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。