tf.stop_gradient

Остановка вычисления градиента.

tf.stop_gradient(
    input, name=None
)

При выполнении в графе этот оператор выводит свой входной тензор без изменений.

При построении операторов для вычисления градиентов этот оператор предотвращает учет вклада его входных данных. Обычно генератор градиентов добавляет в граф операторы для вычисления производных указанной «потери», рекурсивно определяя входящие данные, которые повлияли на её вычисление. Если вы вставляете этот оператор в граф, его входы маскируются от генератора градиента. Они не учитываются при вычислении градиентов.

Это полезно всякий раз, когда вам нужно вычислить значение с помощью TensorFlow, но нужно предположить, что значение является константой. Некоторые примеры включают:

• Алгоритм EM, где M-этап не должен включать обратное распространение через вывод E-этапа.
• Обучение контрастирующим расхождениями машин Больцмана, где при дифференцировании функции энергии обучение не должно обратного распространяться через граф, который генерировал выборки из модели.
• Обучение с помощью антагонистов, где обратное распространение не должно происходить через процесс генерации антагонистических примеров.