昇腾社区首页
中文
注册

MovingSumWithSigmoid

功能

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

输入

  • alpha:输入Tensor,shape=[bst],数据类型支持float16、float。
  • energy:输入Tensor,shape=[bst],数据类型支持float16、float。
  • offset:输入Tensor,shape=[2 * batch_size],数据类型:int32。
  • dec_data:输入Tensor,shape=[bs, feature_dim],数据类型支持float16、float。

属性

ksize:数据类型为int。

输出

y:输出Tensor,shape=[bs, bt],数据类型支持float16、float。

约束与限制

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

支持的ONNX版本

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