MovingSumWithSigmoid

功能

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

输入

三个输入：

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

energy：tensor，1D，shape=[bst]，数据类型：float16、float。

offset：tensor，1D，shape=[2 * batch_size]，数据类型：int32。由beam_num（[b_1,b_2,...,b_batch_size]）和frame_num（[f_1,f_2,...,f_batch_size]）两个数组拼接而成。满足约束b_1 x f_1 + b_2 x f_2 + ... + b_batch_size x f_batch_size = bst。

输出

一个输出：

y：tensor，2D，shape=[bs, bt]，数据类型：float16、float。其中，bs = sum(beam_num)， bt = sum(frame_num)。

属性

ksize：必选，数据类型：int。

约束

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

该算子仅支持Atlas 推理系列产品芯片平台。

支持的ONNX版本

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