数据处理方法及装置、电子设备及存储介质制造方法及图纸

本公开实施例提供一种数据处理方法及装置、电子设备及存储介质，其中，数据处理方法包括：将待处理的第一数据在时间维度与通道维度上合并得到第二数据；基于所述第二数据对应的输出时间维度的大小，对所述第二数据执行卷积和/或反卷积计算；和/或，将所述第一数据在时间维度与批量维度上合并得到第三数据；基于所述第三数据对应的时间维度池化核的大小以及输出时间维度的大小，对所述第三数据执行池化计算。如此，通过时间维度与通道维度的合并，减少待处理数据的维度数量，提高神经网络的数据处理效率和性能。处理效率和性能。处理效率和性能。

【技术实现步骤摘要】
数据处理方法及装置、电子设备及存储介质


[0001]本公开涉及深度学习领域，尤其涉及一种数据处理方法及装置、电子设备及存储介质。

技术介绍

[0002]3D模型对于时间(或深度)方向的信息很敏感，需要对时间维度信息提取并计算处理。但相关技术中由于很多人工智能(Artificial Intelligence，AI)加速器指令并不能直接支持3D形状计算，需要做大量的额外数据格式转换导致深度学习编译器不能很好地对3D模型做编译优化，使得这类网络性能很差。

技术实现思路

[0003]本公开实施例提供一种数据处理方法及装置、电子设备及存储介质。
[0004]本公开实施例第一方面提供一种数据处理方法，包括：
[0005]将待处理的第一数据在时间维度与通道维度上合并得到第二数据；
[0006]基于所述第二数据对应的输出时间维度的大小，对所述第二数据执行卷积和/或反卷积计算；
[0007]和/或，将所述第一数据在时间维度与批量维度上合并得到第三数据；
[0008]基于所述第三数据对应的时间维度池化核的大小以及输出时间维度的大小，对所述第三数据执行池化计算。
[0009]基于上述方案，所述第一数据包含的维度数量为3。
[0010]基于上述方案，所述基于所述第二数据对应的输出时间维度的大小，对所述第二数据执行卷积和/或反卷积计算，包括：
[0011]对所述第二数据执行第一次数的卷积和/或反卷积计算；所述第一次数等于所述第二数据对应的输出时间维度的大小。/>[0012]基于上述方案，所述将待处理的第一数据在时间维度与通道维度上合并得到第二数据，包括：
[0013]将待处理的第一数据的时间维度卷积核的大小与输入通道维度合并，得到第二数据。
[0014]基于上述方案，所述基于所述第三数据对应的时间维度池化核的大小以及输出时间维度的大小，对所述第三数据执行池化计算，包括：
[0015]对所述第三数据执行第二次数的池化计算；所述第二次数等于所述第三数据对应的输出时间维度的大小；
[0016]所述池化计算包括：第三次数的内层计算；所述第三次数等于所述第三数据对应的时间维度池化核的大小。
[0017]基于上述方案，所述池化计算为平均池化计算时，所述池化计算的输出为第三次数的所述内层计算的计算结果平均值；
[0018]所述池化计算为最大值池化计算时，所述池化计算的输出为第三次数的所述内层计算的计算结果中的最大值。
[0019]基于上述方案，所述方法还包括：基于所述第三数据对应的时间维度池化核的大小以及输出时间维度的大小，对所述第三数据执行卷积和/或反卷积计算。
[0020]基于上述方案，所述方法还包括：
[0021]向片外存储位置获取第一数据；
[0022]在计算完成后，通过在合并的时间维度与通道维度上进行拆分，将所述第二数据和/或第三数据恢复为第一数据，并将恢复的所述第一数据传输至所述片外存储位置。
[0023]基于上述方案，所述向片外存储位置获取第一数据，包括：
[0024]基于预设数量的存储块，向片外存储位置获取第一数据；所述预设数量等于所述第一数据的时间维度的大小；
[0025]所述将恢复的所述第一数据传输至所述片外存储位置，包括：
[0026]基于所述预设数量的存储块，将恢复的所述第一数据传输至所述片外存储位置。
[0027]本公开实施例第二方面提供一种数据处理装置，所述装置包括：
[0028]合并单元，用于将待处理的第一数据在时间维度与通道维度上合并得到第二数据，和/或，将所述第一数据在时间维度与批量维度上合并得到第三数据；
[0029]第一计算单元，用于基于所述第二数据对应的输出时间维度的大小，对所述第二数据执行卷积和/或反卷积计算；
[0030]第二计算单元，用于基于所述第三数据对应的时间维度池化核的大小以及输出时间维度的大小，对所述第三数据执行池化计算。
[0031]基于上述方案，所述第一数据包含的维度数量为3。
[0032]基于上述方案，所述第一计算单元，具体用于：
[0033]对所述第二数据执行第一次数的卷积和/或反卷积计算；所述第一次数等于所述第二数据对应的输出时间维度的大小。
[0034]基于上述方案，所述合并单元，具体用于：
[0035]将待处理的第一数据的时间维度卷积核的大小与输入通道维度合并，得到第二数据。
[0036]基于上述方案，所述第二计算单元，具体用于：
[0037]对所述第三数据执行第二次数的池化计算；所述第二次数等于所述第三数据对应的输出时间维度的大小；
[0038]所述池化计算包括：第三次数的内层计算；所述第三次数等于所述第三数据对应的时间维度池化核的大小。
[0039]基于上述方案，所述池化计算为平均池化计算时，所述池化计算的输出为第三次数的所述内层计算的计算结果平均值；
[0040]所述池化计算为最大值池化计算时，所述池化计算的输出为第三次数的所述内层计算的计算结果中的最大值。
[0041]基于上述方案，所述第一计算单元还用于：基于所述第三数据对应的时间维度池化核的大小以及输出时间维度的大小，对所述第三数据执行卷积和/或反卷积计算。
[0042]基于上述方案，所述装置还包括：
[0043]处理单元，用于向片外存储位置获取第一数据；在计算完成后，通过在合并的时间维度与通道维度上进行拆分，将所述第二数据和/或第三数据恢复为第一数据，并将恢复的所述第一数据传输至所述片外存储位置。
[0044]基于上述方案，所述处理单元，具体用于：
[0045]基于预设数量的存储块，向片外存储位置获取第一数据；所述预设数量等于所述第一数据的时间维度的大小；
[0046]基于所述预设数量的存储块，将恢复的所述第一数据传输至所述片外存储位置。
[0047]本公开实施例第三方面提供一种电子设备，包括：
[0048]用于存储处理器可执行指令的存储器；
[0049]处理器，与所述存储器连接；
[0050]其中，所述处理器被配置为执行如前述任意技术方案提供的数据处理方法。
[0051]本公开实施例第四方面提供一种非临时性计算机可读存储介质，当所述存储介质中的指令由计算机的处理器执行时，使得计算机能够执行如前述任意技术方案提供的数据处理方法。
[0052]本公开的实施例提供的技术方案可以包括以下有益效果：
[0053]本公开实施例中提供的数据处理方法包括：将待处理的第一数据在时间维度与通道维度上合并得到第二数据；基于所述第二数据对应的输出时间维度的大小，对所述第二数据执行卷积和/或反卷积计算；和/或，将所述第一数据在时间维度与批量维度上合并得到第三数据；基于所述第三数据对应的时间维度池化核的大小以及输出时间维度的大小，对所述第三数据执行池化计算。如此，通过时间维度与本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种数据处理方法，其特征在于，所述方法包括：将待处理的第一数据在时间维度与通道维度上合并得到第二数据；基于所述第二数据对应的输出时间维度的大小，对所述第二数据执行卷积和/或反卷积计算；和/或，将所述第一数据在时间维度与批量维度上合并得到第三数据；基于所述第三数据对应的时间维度池化核的大小以及输出时间维度的大小，对所述第三数据执行池化计算。2.根据权利要求1所述的方法，其特征在于，所述第一数据包含的维度数量为3。3.根据权利要求1所述的方法，其特征在于，所述基于所述第二数据对应的输出时间维度的大小，对所述第二数据执行卷积和/或反卷积计算，包括：对所述第二数据执行第一次数的卷积和/或反卷积计算；所述第一次数等于所述第二数据对应的输出时间维度的大小。4.根据权利要求1至3任一项所述的方法，其特征在于，所述将待处理的第一数据在时间维度与通道维度上合并得到第二数据，包括：将待处理的第一数据的时间维度卷积核的大小与输入通道维度合并，得到第二数据。5.根据权利要求1所述的方法，其特征在于，所述基于所述第三数据对应的时间维度池化核的大小以及输出时间维度的大小，对所述第三数据执行池化计算，包括：对所述第三数据执行第二次数的池化计算；所述第二次数等于所述第三数据对应的输出时间维度的大小；所述池化计算包括：第三次数的内层计算；所述第三次数等于所述第三数据对应的时间维度池化核的大小。6.根据权利要求5所述的方法，其特征在于，所述池化计算为平均池化计算时，所述池化计算的输出为第三次数的所述内层计算的计算结果平均值；所述池化计算为最大值池化计算时，所述池化计算的输出为第三次数的所述内层计算的计算结果中的最大值。7.根据权利要求1所述的方法，其特征在于，所述方法还包括：基于所述第三数据对应的时间维度池化核的大小以及输出时间维度的大小，对所述第三数据执行卷积和/或反卷积计算。8.根据权利要求1所述的方法，其特征在于，所述方法还包括：向片外存储位置获取第一数据；在计算完成后，通过在合并的时间维度与通道维度上进行拆分，将所述第二数据和/或第三数据恢复为第一数据，并将恢复的所述第一数据传输至所述片外存储位置。9.根据权利要求8所述的方法，其特征在于，所述向片外存储位置获取第一数据，包括：基于预设数量的存储块，向片外存储位置获取第一数据；所述预设数量等于所述第一数据的时间维度的大小；所述将恢复的所述第一数据传输至所述片外存储位置，包括：基于所述预设数量的存储块，将恢复的所述第一数据传输至所述片外存储位置。10.一种数据处理装置，其特征在于，所述装置包括：合并单元，用于将待处理的第一数据在时间维度与通道维...

【专利技术属性】
技术研发人员：陈学儒，蒋国跃，
申请(专利权)人：北京算能科技有限公司，
类型：发明
国别省市：