返回题库
#4简单

实现数值稳定的 Softmax

基础算子Attention

题目描述

实现数值稳定版 softmax。给定长度 n 的输入 x,输出 out

减去最大值是为了避免 exp 溢出。

要求

  • 实现右侧代码模板中给出的函数,签名以模板为准

示例

示例 1

输入:
x = [1.0, 2.0, 3.0]
输出:
[0.0900, 0.2447, 0.6652]

示例 2

输入:
x = [0.0, 0.0]
输出:
[0.5, 0.5]

讨论区(0)

还没有评论,来做第一个发言的人吧

Python · solution.py
编辑器加载中…

运行结果

点击「调试」(只跑样例)或「提交」(跑全部用例)查看结果