Python tf.data.experimental.Reducer用法及代碼示例

reducer 用於減少一組元素。

用法

tf.data.experimental.Reducer(
    init_func, reduce_func, finalize_func
)

reducer 表示為三個函數的元組：

• init_func - 定義初始值：key => 初始狀態
• reducer_func - 對具有相同鍵的值執行的操作：(舊狀態，輸入)=> 新狀態
• finalize_func - 最後返回的值：狀態 => 結果

例如，

def init_func(_):
  return (0.0, 0.0)

def reduce_func(state, value):
  return (state[0] + value['features'], state[1] + 1)

def finalize_func(s, n):
  return s / n

reducer = tf.data.experimental.Reducer(init_func, reduce_func, finalize_func)