[发明专利]一种数据流图处理方法、装置、设备以及可读存储介质

权利要求书

1.一种数据流图处理方法，其特征在于，包括：

获取原始数据流图，解析所述原始数据流图；

若解析出所述原始数据流图包括至少两个逐元素操作节点、待聚合操作节点以及张量形变节点，则检测所述张量形变节点在所述原始数据流图中的位置；

若所述张量形变节点位于目标逐元素操作节点和所述待聚合操作节点之间，则根据所述目标逐元素操作节点在所述原始数据流图中的位置，更新所述张量形变节点在所述原始数据流图中的位置，得到更新数据流图；所述至少两个逐元素操作节点包括所述目标逐元素操作节点；所述更新数据流图中的所述目标逐元素操作节点与所述待聚合操作节点相连接；

在所述更新数据流图中，对所述至少两个逐元素操作节点以及所述待聚合操作节点进行节点聚合优化，得到数据优化流图。

2.根据权利要求1所述的方法，其特征在于，所述至少两个逐元素操作节点之间为并列关系；

所述若所述张量形变节点位于目标逐元素操作节点和所述待聚合操作节点之间，则根据所述目标逐元素操作节点在所述原始数据流图中的位置，更新所述张量形变节点在所述原始数据流图中的位置，得到更新数据流图，包括：

若所述张量形变节点位于目标逐元素操作节点和所述待聚合操作节点之间，且剩余逐元素操作节点均与所述待聚合操作节点相连接，则变换所述张量形变节点与所述目标逐元素操作节点之间的位置，得到更新数据流图；所述剩余逐元素操作节点为所述至少两个逐元素操作节点中，除所述目标逐元素操作节点以外的节点。

3.根据权利要求2所述的方法，其特征在于，所述待聚合操作节点为合并操作节点；

所述在所述更新数据流图中，对所述至少两个逐元素操作节点以及所述待聚合操作节点进行节点聚合优化，得到数据优化流图，包括：

在所述更新数据流图中，若所述目标逐元素操作节点与所述剩余逐元素操作节点的数据操作类型相同，则将所述目标逐元素操作节点以及所述剩余逐元素操作节点进行聚合，得到第一聚合逐元素操作节点；所述第一聚合逐元素操作节点的数据操作类型与所述剩余逐元素操作节点的数据操作类型一致；

将所述第一聚合逐元素操作节点与所述合并操作节点之间的位置进行调换，得到数据优化流图；在所述数据优化流图中，所述合并操作节点的输出张量是所述第一聚合逐元素操作节点的输入张量。

4.根据权利要求3所述的方法，其特征在于，还包括：

根据所述数据优化流图配置数据处理模型；

获取第一待合并张量，将所述第一待合并张量输入至所述数据处理模型；

通过所述数据处理模型中的所述张量形变节点，对所述第一待合并张量的形状进行转换，生成所述第一待合并张量对应的转换张量，将所述转换张量输入所述合并操作节点；

获取第二待合并张量，将所述第二待合并张量输入所述合并操作节点；

通过所述合并操作节点对所述第一待合并张量与所述第二待合并张量进行合并，生成合并张量；

将所述合并张量输入所述第一聚合逐元素操作节点，通过所述第一聚合逐元素操作节点对所述合并张量进行逐元素操作处理，生成目标输出张量。

5.根据权利要求2所述的方法，其特征在于，所述待聚合操作节点为分裂操作节点；

所述在所述更新数据流图中，对所述至少两个逐元素操作节点以及所述待聚合操作节点进行节点聚合优化，得到数据优化流图，包括：

在所述更新数据流图中，若所述目标逐元素操作节点与所述剩余逐元素操作节点的数据操作类型相同，则将所述目标逐元素操作节点以及所述剩余逐元素操作节点进行聚合，得到第二聚合逐元素操作节点；所述第二聚合逐元素操作节点的数据操作类型与所述剩余逐元素操作节点的数据操作类型一致；

将所述第二聚合逐元素操作节点与所述分裂操作节点之间的位置进行调换，得到数据优化流图；在所述数据优化流图中，所述第二聚合逐元素操作节点的输出张量是所述分裂操作节点的输入张量。