功能描述

src

输入Tensor。

dst

输出Tensor。

c

softmax运算所沿维度。

最外维度。

最内维度。

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