tf.AggregationMethod
| Просмотреть исходный код на GitHub |
Класс, перечисляющий методы агрегирования, используемые для объединения градиентов.
Вычисление частных производных может потребовать агрегирования вкладов градиента. Этот класс перечисляет различные методы объединения градиентов в графе.
Следующие методы агрегирования являются частью стабильного 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