下载
中文
注册

DynSeqOuter

功能

按照偏移量求和。

输入

四个输入:

x1:tensor,shape=[bs, feature_dim],数据类型:float16、float。bs: 单批数据个数,feature_dim:特征维度。

x2:tensor,shape=[bs, feature_dim],数据类型:float16、float。bs: 单批数据个数,feature_dim:特征维度。

seq_len1:tensor,shape=[batch_size]且bs=sum(seq_len1),数据类型:int32。

seq_len2:tensor,shape=[batch_size]且bt=sum(seq_len2),数据类型:int32。

输出

一个输出:

y:tensor,shape=[bst, feature_dim]且bst=sum(seq_len1 * seq_len2),数据类型:float16、float。

约束

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

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

支持的ONNX版本

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