[发明专利]一种张量转置方法、装置、计算机及存储介质

说明书

本发明实施例公开了一种张量转置方法、装置、计算机及存储介质。其中方法包括：获取待处理的第一张量数据和第一张量数据的期望维度序列；当第一张量数据的维度序列和期望维度序列不同时，根据转置处理顺序、第一张量数据的维度序列和期望维度序列确定当前处理的张量维度；根据当前处理的张量维度和张量转换规则，将第一张量数据转换为第一预设维度数量的第二张量数据；根据期望维度序列对第二张量数据中与第一张量数据中的当前处理的张量维度相对应的维度数据进行转置处理；将处理后的第二张量数据进行张量反转换为第三张量数据，其中，第三张量数据与第一张量数据的维度数量相同。基于上述技术特征，实现对任意维度的张量数据进行转置处理。

技术领域

本发明实施例涉及数据处理技术，尤其涉及一种张量转置方法、装置、计算机及存储介质。

背景技术

随着技术的进步，每天都有越来越多的数据被创建和分析。机器学习技术(例如深度学习和卷积神经网络)，作为分析这些大量数据的重要方法，正变得日益重要。

张量是一个数学概念，包括标量、向量和线性算子，可通过坐标系统来表达，标记为张量的数字。张量被广泛应用于深度神经网络领域，用于表示神经元的权重信息。张量转置是神经网络中常用的一种张量操作，用于在逻辑上调换张量各维度的维度顺序，当神经网络中张量格式与原始张量数据格式不同时，可通过张量转置的处理方式将原始张量数据转换为神经网络中需求的张量格式。

目前，进行张量转置处理一般是采用直接内存访问设备(Direct Memory Access，DMA)实现，但是上述硬件仅支持有限维度的张量转置，无法实现其他任意维度的张量转置，尤其是维度大于硬件可支持的有限维度的张量数据。

发明内容

本发明提供一种张量转置方法、装置、计算机及存储介质，以实现对任意维度的张量数据进行转置处理。

第一方面，本发明实施例提供了一种张量转置方法，包括：

获取待处理的第一张量数据和所述第一张量数据的期望维度序列；

当所述第一张量数据的维度序列和所述期望维度序列不同时，根据转置处理顺序、所述第一张量数据的维度序列和所述期望维度序列确定当前处理的张量维度；

根据所述当前处理的张量维度和张量转换规则，将所述第一张量数据转换为第一预设维度数量的第二张量数据；

根据所述期望维度序列对所述第二张量数据中与所述第一张量数据中的当前处理的张量维度相对应的维度数据进行转置处理；

将处理后的第二张量数据进行张量反转换为第三张量数据，其中，所述第三张量数据与所述第一张量数据的维度数量相同。

第二方面，本发明实施例还提供了一种张量转置装置，该装置包括：

数据获取模块，用于获取待处理的第一张量数据和所述第一张量数据的期望维度序列；

当前处理的张量维度确定模块，用于当所述第一张量数据的维度序列和所述期望维度序列不同时，根据转置处理顺序、所述第一张量数据的维度序列和所述期望维度序列确定当前处理的张量维度；

第二张量数据确定模块，用于根据所述当前处理的张量维度和张量转换规则，将所述第一张量数据转换为第一预设维度数量的第二张量数据；

数据转置模块，用于根据所述期望维度序列对所述第二张量数据中与所述第一张量数据中的当前处理的张量维度相对应的维度数据进行转置处理；

第三张量数据确定模块，用于将处理后的第二张量数据进行张量反转换为第三张量数据，其中，所述第三张量数据与所述第一张量数据的维度数量相同。

第三方面，本发明实施例还提供了一种计算机可读存储介质，其上存储有计算机程序，该程序被处理器执行时实现如本发明任意实施例提供的张量转置方法。