LogSoftMax
功能
对输入tensor计算logsoftmax值。
输入
一个输入:
x:一个tensor,数据类型:float16、float。
输出
一个输出:
y:一个tensor,数据类型和shape与输入一致。
属性
axis:数据类型为int;指定计算的轴,取值范围:[-r, r-1],r为输入的秩。
约束
float16 不支持onnx v18。
支持的ONNX版本
Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18
父主题: 支持ONNX算子清单