tf.raw_ops.StopGradient

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

tf.raw_ops.StopGradient(
    input, name=None
)

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

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

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

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