Spec-Zone.ru › TensorFlow 2.4

tf.AggregationMethod

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

Класс, перечисляющий методы агрегирования, используемые для объединения градиентов.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительных сведений.

tf.compat.v1.AggregationMethod

Вычисление частных производных может потребовать агрегирования вкладов градиента. Этот класс перечисляет различные методы объединения градиентов в графе.

Следующие методы агрегирования являются частью стабильного API для агрегирования градиентов:

  • ADD_N: Все члены градиента суммируются в рамках одной операции с использованием операции "AddN" (см. tf.add_n). Этот метод требует, чтобы все градиенты были готовы и буферизованы отдельно в памяти перед выполнением агрегирования.
  • DEFAULT: Метод агрегирования по умолчанию, выбранный системой.

Следующие методы агрегирования являются экспериментальными и могут не поддерживаться в будущих выпусках:

  • EXPERIMENTAL_TREE: Члены градиента суммируются парами с использованием операции "AddN". Этот метод суммирования градиентов может снизить производительность, но может улучшить использование памяти, поскольку градиенты могут быть освобождены раньше.
Переменные класса
ADD_N 0
DEFAULT 0
EXPERIMENTAL_ACCUMULATE_N 2
EXPERIMENTAL_TREE 1

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/AggregationMethod

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API