鲲鹏社区首页
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助

功能描述

算子功能

沿数据的某一维执行Softmax函数运算。

算子公式

表1 公式参数说明

参数

参数说明

输入Tensor。

dst

输出Tensor。

softmax运算所沿维度。

最外维。

最内维。

用于产生数值稳定输出结果的系数,通过以下公式计算,其中ic是从最外围到最内维的所有中间维度。