Флаг говорит loss, что вход — сырые logits, а не вероятности; внутри применится стабильный sigmoid/softmax.
Разбор
- Предпочтительнее численно, чем Softmax+loss без from_logits.
- SparseCategoricalCrossentropy(from_logits=True) — частый паттерн.
- На инференсе вероятности получают отдельным Softmax.
- Несоответствие флага ломает обучение.
Итог
from_logits связывает формат выхода модели со стабильным loss.