tf.compat.v1.data.experimental.StatsAggregator

Состоятельный ресурс, который агрегирует статистику из одного или нескольких итераторов.

tf.compat.v1.data.experimental.StatsAggregator()

Для записи статистики используйте одну из пользовательских функций преобразования, определённых в этом модуле, при определении вашего tf.data.Dataset. Вся статистика будет агрегироваться StatsAggregator , связанной с конкретным итератором (см. ниже). Например, чтобы записать задержку создания каждого элемента при итерации по набору данных:

dataset = ...
dataset = dataset.apply(tf.data.experimental.latency_stats("total_bytes"))

Чтобы связать StatsAggregator с объектом tf.data.Dataset, используйте следующий шаблон:

aggregator = tf.data.experimental.StatsAggregator()
dataset = ...

# Apply `StatsOptions` to associate `dataset` with `aggregator`.
options = tf.data.Options()
options.experimental_stats.aggregator = aggregator
dataset = dataset.with_options(options)

Чтобы получить протокольный буфер с кратким описанием текущей агрегированной статистики, используйте StatsAggregator.get_summary() тензор. Самый простой способ сделать это — добавить возвращённый тензор в tf.GraphKeys.SUMMARIES коллекцию, чтобы описания были включены в любые существующие описания.

aggregator = tf.data.experimental.StatsAggregator()
# ...
stats_summary = aggregator.get_summary()
tf.compat.v1.add_to_collection(tf.GraphKeys.SUMMARIES, stats_summary)

Примечание: Этот интерфейс является экспериментальным и ожидается, что он изменится. В частности, ожидается добавить другие реализации StatsAggregator, которые предоставляют разные способы экспорта статистики, и добавить больше типов статистики.

Методы

get_summary

Просмотреть исходный код

get_summary()

Возвращает строку tf.Tensor, которая суммирует агрегированную статистику.

Возвращённый тензор будет содержать сериализованный tf.compat.v1.summary.Summary протокольный буфер, который можно использовать с стандартными средствами протоколирования TensorBoard.