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