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

MovingSumWithSigmoid

功能

对输入alpha进行滑窗求和,输入energy做Sigmoid计算,两者乘积生成输出。

输入

三个输入:

alpha:tensor,1D,shape=[bst],数据类型:float16、float。

energy:tensor,2D,shape=[bst],数据类型:float16、float。

offset:tensor,1D,shape=[2 * batch_size],数据类型:int32。

输出

一个输出:

y:tensor,2D,shape=[bs, bt],数据类型:float16、float。

属性

ksize:必选,数据类型:int。

约束

该算子为特定用户场景定制,非相关场景不建议使用。

该算子仅支持Atlas 推理系列产品(Ascend 310P AI处理器)芯片平台。

支持的ONNX版本

Opset v8/v9/v10/v11/v12/v13/v14/v15/v16

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

当前产品无相关内容

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