[发明专利]张量切分方法、装置、芯片及介质

权利要求书

1.一种张量切分方法，用于跨步切分，其特征在于，包括：

获取初始张量和切分参数；

根据所述初始张量和所述切分参数对待切分维度上的数据进行重排，得到重排张量，使得所述待切分维度上的待保留数据相邻排布；

根据所述切分参数对所述重排张量进行切分，得到结果张量。

2.根据权利要求1所述的方法，其特征在于，所述切分参数包括：切分步长；

根据所述初始张量和所述切分参数对所述待切分维度上的数据进行重排，得到重排张量，包括：

根据所述待切分维度的维度尺寸和切分步长将所述待切分维度拆分为切分子维度和步长子维度，并根据拆分结果对所述待切分维度上的数据进行重排，得到重排张量。

3.根据权利要求2所述的方法，其特征在于，根据拆分结果对所述待切分维度上的数据进行重排，包括：

依次获取所述待切分维度上的各个数据作为当前处理数据；

将所述当前处理数据排布在目标元素位置上；其中，所述目标元素位置在所述步长子维度上的索引值为所述当前处理数据在所述待切分维度上的索引值除以所述切分步长的余数值；所述目标元素位置在所述切分子维度上的索引值为所述当前处理数据在所述待切分维度上的索引值除以所述切分步长的商值；

返回执行按顺序依次获取所述待切分维度上的一个数据作为当前处理数据的操作，直至处理完所述待切分维度上的所有数据。

4.根据权利要求2所述的方法，其特征在于，所述切分参数还包括切分起始索引值和切分结束索引值；

根据所述切分参数对所述重排张量进行切分，得到结果张量，包括：

根据所述待切分维度的切分起始索引值，在所述步长子维度上对所述重排张量进行第一切分，得到一次切分张量；

根据所述待切分维度的切分起始索引值、切分结束索引值以及切分步长，在所述切分子维度上对所述一次切分张量进行第二切分，得到二次切分张量；

重排所述二次切分张量，使所述二次切分张量与所述初始张量的形状匹配，并将重排后的二次切分张量作为所述结果张量。

5.根据权利要求4所述的方法，其特征在于，根据所述待切分维度的切分起始索引值，在所述步长子维度上对所述重排张量进行第一切分，得到一次切分张量，得到一次切分张量，包括：

如果所述待切分维度的切分起始索引值小于所述维度的切分步长，则在所述重排张量中切分出所述步长子维度上索引值为所述切分起始索引值的张量，作为所述一次切分张量；

如果所述待切分维度的切分起始索引值大于等于所述维度的切分步长，则在所述重排张量中切分出所述步长子维度上索引值为目标值的张量，作为所述一次切分张量；其中，所述目标值为所述切分起始索引值除以所述切分步长的余数值。

6.根据权利要求5所述的方法，其特征在于，根据所述待切分维度的切分起始索引值、切分结束索引值以及切分步长，在所述切分子维度上对所述一次切分张量进行第二切分，得到二次切分张量，包括：

根据所述待切分维度的切分起始索引值、切分结束索引值以及切分步长，计算在所述切分子维度上的切分尺寸；

将所述切分起始索引值除以所述切分步长的商值，作为所述切分子维度上的切分起始索引值；

根据所述切分子维度上的切分起始索引值及切分尺寸，在所述切分子维度上对所述一次切分张量进行切分，得到二次切分张量。

7.根据权利要求2所述的方法，其特征在于，根据所述待切分维度的维度尺寸和切分步长将所述待切分维度拆分为切分子维度和步长子维度，包括：

如果所述待切分维度的维度尺寸不能被所述待切分维度的切分步长整除，则增大所述待切分维度的维度尺寸，以使增大后的维度尺寸能够被所述待切分维度的切分步长整除；

根据所述增大后的维度尺寸和切分步长将所述待切分维度拆分为切分子维度和步长子维度；

其中，所述切分子维度的维度尺寸为所述增大后的维度尺寸与所述切分步长的商值，所述步长子维度的维度尺寸为所述切分步长的值。