Я хочу написать собственную функцию потерь для моей нейронной сети в TensorFlow. Мне ясно, как написать простую пользовательскую функцию потерь, но я хочу, чтобы функция потерь зависела от прогнозируемого значения сети. Моя сеть выводит значения от 0 до 1, и если вывод выше 0,95, я хочу применить другую функцию потерь, чем когда вывод меньше 0,95. Проблема, с которой я столкнулся, заключается в том, что предсказанные значения в функции потерь представлены в виде тензора, и я не могу найти решение для доступа к отдельным элементам этого тензора, чтобы увидеть, какую функцию стоимости мне нужно применить. Спасибо за помощь!