Softmax

功能

对输入Tensor应用Softmax函数。

$\text{[math]}$

输入

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

父主题： 支持ONNX算子清单