Softmax

功能

对输入进行softmax

边界

【输入】

一个输入

x:一个tensor,数据类型:float16、float32、double

【输出】

一个输出

y:一个tensor,类型和shape与输入x一致

【属性】

axis:Int,可选,表示进行softmax的方向,默认值为-1,范围为[ -len(x.shape), len(x.shape)-1]

支持的ONNX版本

Opset v8/v9/v10/v11/v12/v13