Я пытаюсь создать собственную функцию потерь, которая использует градиент выходных данных, прогнозируемых моделью (то есть градиент y_pred). Например, функция потерь может пытаться минимизировать среднеквадратичную ошибку между градиентом выходных данных модели (grad (y_pred)) и y_true.
Хотя многие операции, такие как взятие квадрата y_pred внутри пользовательской функции потерь, в порядке, использование градиента в пользовательской функции потерь невозможно. Я понимаю, что это может быть связано с тем, что y_pred является просто ссылкой на вычислительный граф и изначально не имеет никаких значений. Может кто-нибудь, пожалуйста, предложить обходной путь для этого?