Thử thách lập trình Softmax Function

Đề bài

The softmax function turns a list of scores (logits) into probabilities. Language models use it to choose the next word:

softmax(x)_i = e^(x_i) / (e^(x_1) + e^(x_2) + … + e^(x_n))

Large scores make e^x overflow, so first subtract the largest score from every score. The result is mathematically the same.

Implement softmax(logits).

logits: A list of scores.

Return the list of probabilities, in the same order. They are all between 0 and 1 and add up to 1.

1 ≤ length ≤ 1,000

-1,000 ≤ each score ≤ 1,000

Ví dụ 1

Đầu vào:
logits = [1.0, 2.0, 3.0]

Kết quả mong đợi:
expected = [0.090031, 0.244728, 0.665241]

Đầu vào:
std::vector<double> logits = {1.0, 2.0, 3.0};

Kết quả mong đợi:
std::vector<double> expected = {0.090031, 0.244728, 0.665241};

Đầu vào:
double[] logits = new double[] {1.0, 2.0, 3.0};

Kết quả mong đợi:
double[] expected = new double[] {0.090031, 0.244728, 0.665241};

e¹ ≈ 2.718, e² ≈ 7.389 and e³ ≈ 20.086 add up to 30.193, so the probabilities are about [0.090031, 0.244728, 0.665241].

Ví dụ 2

Đầu vào:
logits = [1000.0, 1000.0]

Kết quả mong đợi:
expected = [0.5, 0.5]

Đầu vào:
std::vector<double> logits = {1000.0, 1000.0};

Kết quả mong đợi:
std::vector<double> expected = {0.5, 0.5};

Đầu vào:
double[] logits = new double[] {1000.0, 1000.0};

Kết quả mong đợi:
double[] expected = new double[] {0.5, 0.5};

Computing e^1000 directly overflows. Subtracting the maximum gives [0, 0], and e⁰ / (e⁰ + e⁰) = 0.5 for both.

Thử thách này đã được điều chỉnh cho CodeUtility. Hãy làm theo đề bài và các ràng buộc ở trên.