Spec-Zone.ru › TensorFlow 2.3

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.3/api_docs/python/tf/AggregationMethod

Spec-Zone.ru

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