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)