昇腾社区首页
中文
注册

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