数值稳定的 Softmax

大模型手撕 · 简单 · 基础算子、数值稳定性

实现沿指定维度的 Softmax：

数据范围

在 RagCodes 上用 Python / C++ / Java 在线编写并提交，核心代码模式与 ACM 模式均可评测。