文档
注册
评分
提单
论坛
小AI

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

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词