[发明专利]基于广播机制进行张量计算的方法、装置、芯片及介质

权利要求书

1.一种基于广播机制进行张量计算的方法，其特征在于，包括：

获取第一张量和第二张量，第一张量的维数大于或等于第二张量的维数；

判断第一张量和第二张量是否满足广播机制条件，若满足，则确定与第一张量和第二张量分别对应的第一目标张量和第二目标张量，第一目标张量和第二目标张量的后缘维度的轴长度相符；

通过通用计算硬件单元组，采用广播机制对第一目标张量和第二目标张量进行张量计算，得到张量计算备选结果；

根据张量计算备选结果，确定与第一张量和第二张量对应的目标计算结果。

2.根据权利要求1所述的方法，其特征在于，第一张量和第二张量的后缘维度的轴长度不相符；

确定与第一张量和第二张量分别对应的第一目标张量和第二目标张量，包括：

通过张量处理硬件单元组对第一张量和/或第二张量进行处理，得到第一目标张量和第二目标张量。

3.根据权利要求2所述的方法，其特征在于，通过张量处理硬件单元组对第一张量和/或第二张量进行处理，得到第一目标张量和第二目标张量，包括：

根据第一张量和第二张量中对应维度的维度尺寸，获取满足张量轴移条件的至少一个第一维度，和/或满足张量拼接条件的至少一个第二维度；

通过张量处理硬件单元组，根据第一维度对第一张量和第二张量进行张量轴移处理，和/或对第一张量的第二维度进行与第二张量匹配的张量拼接处理，得到第一目标张量和第二目标张量。

4.根据权利要求3所述的方法，其特征在于，根据第一张量和第二张量中对应维度的维度尺寸，获取满足张量轴移条件的至少一个第一维度，包括：

以第二张量的最高维度为起点，在第二张量中获取维度尺寸不为1的首个维度作为起始维度；

在所述起始维度至最低维度的范围内，比较第一张量和第二张量对应维度的维度尺寸，依次记录维度尺寸不相等且第二张量的维度尺寸为1的第一维度；

通过张量处理硬件单元组，根据第一维度对第一张量和第二张量进行张量轴移处理，包括：

通过张量处理硬件单元组，按照记录顺序将第一张量和第二张量中的各第一维度依次轴移到所述起始维度所在位置。

5.根据权利要求4所述的方法，其特征在于，根据张量计算备选结果，确定与第一张量和第二张量对应的目标计算结果，包括：

通过张量处理硬件单元组，按照与所述张量轴移处理匹配的逆处理过程，对张量计算备选结果进行处理，得到所述目标计算结果。

6.根据权利要求5所述的方法，其特征在于，依次记录维度尺寸不相等且第二张量的维度尺寸为1的第一维度，包括：

在轴移列表中依次记录各所述第一维度；

通过张量处理硬件单元组，按照记录顺序将第一张量和第二张量中的各第一维度依次轴移到所述起始维度所在位置，包括：

通过张量处理硬件单元组，按照轴移列表中记录顺序将第一张量和第二张量中的各第一维度依次轴移到所述起始维度所在位置。

7.根据权利要求6所述的方法，其特征在于，通过张量处理硬件单元组，按照与所述张量轴移处理匹配的逆处理过程，对张量计算备选结果进行处理，得到所述目标计算结果，包括：

通过张量处理硬件单元组，按照第一维度记录顺序的逆序，将张量计算备选结果中所述起始维度依次轴移到各所述第一维度。

8.根据权利要求6所述的方法，其特征在于，通过张量处理硬件单元组，按照与所述张量轴移处理匹配的逆处理过程，对张量计算备选结果进行处理，得到所述目标计算结果，包括：

通过张量处理硬件单元组，按照反向轴移列表中第一维度的记录顺序，将张量计算备选结果中所述起始维度依次轴移到各所述第一维度；

其中，反向轴移列表和轴移列表中第一维度的记录顺序呈逆序。