Код Junior Инференс и генерация спрашивают часто обновлено
Softmax с температурой
Условие
Напишите softmax_t(logits, t): вероятности по логитам с температурой t > 0. Делите логиты на t, для численной устойчивости вычитайте максимум перед экспонентой. Верните список вероятностей, округлённых до 4 знаков (round(p, 4)).
Примеры
softmax_t([1, 2, 3], 1) // [0.09, 0.2447, 0.6652]
softmax_t([1, 2, 3], 0.1) // [0, 0, 1]
softmax_t([1, 2, 3], 10) // [0.3006, 0.3322, 0.3672]
Как подойти
Подсказки: 1
exp(x - max)не переполняется.
Решите в тренажёре
Редактор с автотестами прямо в браузере. Можно решить спокойно, вслух или под таймером, как на собесе.