[发明专利]访问多维张量中的数据

摘要

本申请涉及访问多维张量中的数据。用于处理用来访问N维张量的指令的方法、系统、和装置，所述装置包括多个张量索引元素和多个维度乘子元素，其中，所述维度乘子元素中的每一个维度乘子元素具有对应张量索引元素。所述装置包括一个或多个处理器，所述一个或多个处理器被配置成获得用于访问N维张量的特定元素的指令，其中，所述N维张量具有跨N个维度中的每一个维度布置的多个元素，并且其中，N是等于或大于1的整数；使用所述多个张量索引元素中的一个或多个张量索引元素和所述多个维度乘子元素中的一个或多个维度乘子元素来确定所述特定元素的地址；以及输出指示用于访问所述N维张量的所述特定元素的所确定的地址的数据。

主权项

1.一种用于处理多维张量输入的装置，所述装置包括：/n多个张量索引元素，所述多个张量索引元素包括用于遍历N维张量的多个嵌套循环中的每个嵌套循环的张量索引元素，其中，每个张量索引元素是相应的第一硬件寄存器，所述相应的第一硬件寄存器被配置来存储用于遍历所述N维张量的相应维度的相应嵌套循环的索引值，并且其中，所述N维张量包括跨N个维度中的每一个维度布置的数据元素并且N是等于或大于一的整数，其中，每个索引值是每次该索引值的相应嵌套循环的迭代被执行时更新的值；/n多个维度乘子元素，其中，每个维度乘子元素是被配置来存储所述N维张量的相应维度的乘子值的相应的第二硬件寄存器，其中，每个乘子值是所述N维张量的、该乘子值的相应维度的常数值并且在用于遍历该相应维度的相应嵌套循环的每次迭代中保持恒定；以及/n一个或多个硬件处理器，所述一个或多个硬件处理器被配置来执行由所述一个或多个硬件处理器可执行的指令集中的一个或多个指令，其中，所述一个或多个指令的执行使得所述一个或多个硬件处理器执行包括以下的操作：/n通过以下操作来确定存储器中的、用于存储所述N维张量的数据元素序列的数据值的位置的存储器地址：/n对于所述嵌套循环的内循环的每次迭代：/n对于所述N维张量中的每一个维度，确定以下的乘积：(i)用于遍历该维度的嵌套循环的张量索引元素中存储的索引值以及(ii)该维度的乘子值；以及/n基于所述N维张量中的每一个维度的乘积的总和来确定与所述内循环的该次迭代相对应的数据元素的存储器地址；以及/n输出指示所述N维张量的所述数据元素序列中的每个数据元素的所确定的存储器地址的数据。/n