tf.raw_ops.StopGradient

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

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

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

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

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

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