tf.data.experimental.Reducer

Редуктор используется для сокращения набора элементов.

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

Редуктор представлен как кортеж из трех функций:

• init_func - для определения начального значения: ключ => начальное состояние
• 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)