买专利卖专利找龙图腾，真高效！ 查专利查商标用IPTOP,全免费！专利年费监控用IP管家,真方便！

申请/专利权人：广东工业大学

摘要：本发明公开了一种基于结构注意力的代码摘要生成方法及装置，具体为：S1、将原始抽象语法树的节点分裂为类型节点和值节点，得到分裂后的抽象语法树；S2、基于分裂后的抽象语法树，得到原始边、构建控制流边、数据流边，并设计原始边的邻接矩阵；S3、对原始边的邻接矩阵与控制流以及原始边的邻接矩阵与数据流分别进行联合运算，得到基于结构的注意力分数矩阵；S4、利用基于结构的注意力得分矩阵、前序遍历的序列、边的关系矩阵和层级矩阵，构建基于结构注意力的代码摘要模型；S5，利用构建的基于结构注意力的代码摘要模型，生成代码摘要，本发明能更加全面地学习到源代码的特征，从而更好的进行代码摘要的生成。

主权项：1.一种基于结构注意力的代码摘要生成方法，其特征在于，包括以下步骤：S1、将源代码的代码片段解析为原始抽象语法树，并将原始抽象语法树的节点分裂为类型节点和值节点，得到分裂后的抽象语法树；S2、基于分裂后的抽象语法树，为抽象语法树中的终端值节点、类型节点和值节点分别构建连接边，得到原始边；基于抽象语法树中的控制流和数据流，构建控制流边、数据流边，设计原始边的邻接矩阵；为抽象语法树中的叶子节点构建连接边；S3、对原始边的邻接矩阵与控制流以及原始边的邻接矩阵与数据流分别进行联合运算，初步得到Origin-Control矩阵和Origin-Dataflow矩阵，并通过分别计算Origin-Control矩阵和Origin-Dataflow矩阵的二次幂，最终得到矩阵中每个元素代表从一个节点到另一个节点路径数为2的两个基于结构的注意力分数矩阵；基于原始边、控制流边和数据流边构建边关系矩阵；对抽象语法树中的所有节点进行前序遍历，并为抽象语法树中的所有节点创建one-hot矩阵；S4、利用基于结构的注意力得分矩阵、前序遍历的序列、边的关系矩阵和层级矩阵，构建基于结构注意力的代码摘要模型；所述代码摘要模型包括GraphSAGE编码器、Transformer编码器和融合解码器；基于GraphSAGE编码器学习前序遍历的抽象语法树序列中局部视角的特征；基于Transformer编码器将代码片段中每个词元的嵌入添加到抽象语法树序列中相应节点的嵌入后，再与层级矩阵拼接；在融合解码器中部署的双交叉注意机制，并基于融合解码器分别拾取GraphSAGE编码器和Transformer编码器所学习到前序遍历的抽象语法树序列的不同视角的特征；S5，利用构建的基于结构注意力的代码摘要模型，生成代码摘要。

全文数据：

权利要求：

百度查询： 广东工业大学 一种基于结构注意力的代码摘要生成方法及装置