题解 | Log Softmax函数的实现

import numpy as np

from pandas import array
def log_softmax(scores: list) -> np.ndarray:
    scores = np.array(scores)
    max_score= np.max(scores)
    exp_scores = np.exp(scores - max_score)
    sum_exp_scores = np.sum(exp_scores)
    return np.log(exp_scores / sum_exp_scores)

if __name__ == "__main__":
    scores = eval(input())
    print(log_softmax(scores))