Problem · Array
Handwritten Softmax
Problem statement
Given a nonempty double array logits, return its softmax probability array in the same order.
Let m be the maximum logit. For every index i, compute:
The problem statement continues
ProExamples
Example 1
logits = [1.0,2.0,3.0]return = [0.09003057317038046,0.24472847105479764,0.6652409557748218]After subtracting the maximum 3, normalize [exp(-2), exp(-1), 1] by their sum.
FastPrep Pro
Reported in 1 Amazon interview this weekUnlock this recently reported problem
FastPrep Pro gives you full access to interview problems reported within the last week.
- Full problem statement and constraints
- 2 more worked examples, explained
- Guided hints and editorial
- Run your code on real test cases
$9/month
Pro subscription, billed yearly — or $19 month-to-month. Cancel anytime.
Free plan — 2 of 2 free unlocks used this week