DynamicAUGRU(Attention-based Update Gate GRU,由注意力分数调节更新门的门控循环单元)用于处理用户行为等序列数据。ops-nn 仓库当前无该算子实现,需要新增以支持 Ascend 950(arch35)及图模式动态 Shape 推理。
[T,B,I]
[T,B,H]
seq_length
[B]
bias_input
bias_hidden
init_h
docs/zh/op_list.md
背景
DynamicAUGRU(Attention-based Update Gate GRU,由注意力分数调节更新门的门控循环单元)用于处理用户行为等序列数据。ops-nn 仓库当前无该算子实现,需要新增以支持 Ascend 950(arch35)及图模式动态 Shape 推理。
需求
[T,B,I]与权重、注意力分数输入,7 个[T,B,H]输出),实现图模式推理与动态 Shape 推导。seq_length支持 INT32[B](有效序列长度)与 FLOAT16[T,B,H](逐元素掩码)两种模式;bias_input/bias_hidden/init_h为可选输入,缺省按零处理。docs/zh/op_list.md注册。验收标准