Softmax

功能

对输入Tensor应用Softmax函数。

输入

input:输入Tensor,数据类型支持float16、float、double。

属性

axis(可选):int,用于指定进行Softmax计算的轴,默认值为-1。

输出

output:输出Tensor,数据类型和shape与input保持一致。

限制与约束

无。

支持的ONNX版本

Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18