Реализация мягкого argmax в tensorflow

#python-3.x #tensorflow #tensorflow2.0 #gradient #python-3.x #тензорный поток #тензорный поток 2,0 #градиент Вопрос: Я пытаюсь реализовать мягкую операцию argmax в Tensorflow. Я хочу, чтобы при прямом проходе был нормальный argmax, а…