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
【约束】
该算子为特定用户场景定制,非相关场景不建议使用。
支持的ONNX版本
Opset v8/v9/v10/v11/v12/v13/v14/v15/v16
父主题: 支持ONNX算子清单