Softmax多分类激活函数，将多维（最大8维）Tensor数据在指定轴上映射到0到1之间，且非softmax轴数值之和为1。

图1 SoftmaxOperation

struct SoftmaxParam {
    SVector<int64_t> axes;
};

成员名称	描述
axes	指定轴（维度），axes可以支持多个轴上进行处理。 axes不能为空，当指定多个轴时，多个轴之间必须连续且从小到大排列。 axes的元素要求大于或等于-1且小于输入x的维度。

参数	维度	数据类型	格式
x	[-1,…,-1] -1表示当前维度的大小没有约束。	float16/float	ND

参数

维度

数据类型

格式

x

[-1,…,-1]

-1表示当前维度的大小没有约束。

float16/float

ND

参数	维度	数据类型	格式
output	[-1,…,-1] -1表示当前维度的大小没有约束。	float16/float	ND

参数

维度

数据类型

格式

output

[-1,…,-1]

-1表示当前维度的大小没有约束。

float16/float

ND