SoftmaxV2

产品支持情况

产品	是否支持
Atlas A2 训练系列产品/Atlas 800I A2 推理产品	√

功能说明

• 算子功能：对输入张量 x 在指定维度 dim 上执行 Softmax 运算，将该维度上的数值归一化为概率分布。
• 计算公式：

$$y_i=\frac{\exp (x_i-m)}{{\sum }_j\exp (x_j-m)},m=\underset{j}{\max }(x_j)$$

• 其中，Softmax 的 max 和 sum 运算均在指定维度 dim 上进行，其余维度保持不变。
• 功能说明：支持在4维合法维度上进行 Softmax 计算，内部采用减去最大值（这个最大值最小为0）的方式提升数值稳定性，输出张量 shape 与输入张量 shape 完全一致

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入张量	需要进行维度置换的输入张量。	见下方	ND
dim	输入数组	整型，代表指定的维度。取值需在[0，3]范围内。	INT32	-
out	输出	维度为4维，shape和原selfx的一致，dtype需要与selfx一致。	同x	ND

• Atlas A2 训练系列产品/Atlas 800I A2 推理产品：数据类型支持FLOAT、FLOAT16

约束说明

• dim取值只能为正

调用说明

调用方式	样例代码	说明
aclnn接口	test_softmax_v2v	通过[aclnnSoftmaxV2v]接口方式调用softmax_v2v算子。