题解 | softmax激活函数实现

import numpy as np
import math


def softmax(scores: list[float]) -> list[float]:
    exp_scores = np.exp(scores)
    probabilities = exp_scores/np.sum(exp_scores)
    return np.round(probabilities,4).tolist()

if __name__ == "__main__":
    scores = np.array(eval(input()))
    print(softmax(scores))