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 |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/AggregationMethod