tf.data.experimental.set_stats_aggregator

tf.data.experimental.set_stats_aggregator(stats_aggregator)

Defined in tensorflow/python/data/experimental/ops/stats_ops.py.

Set the given stats_aggregator for aggregating the input dataset stats.

Args:

Returns:

A Dataset transformation function, which can be passed to tf.data.Dataset.apply.