[发明专利]处理张量数据的方法和装置

说明书

公开了一种处理张量数据的方法和装置。该方法可以包括：存储多个指令序列，每个指令序列用于处理与该指令序列相对应的数量的张量；针对输入张量的每个感兴趣区域，从输入张量中确定与该感兴趣区域对应的局部张量；以及针对所获取的每第一数量的局部张量，执行用于处理第一数量的张量的第一指令序列。由此，能够减少数据的传输量，并提升执行速度。

技术领域

本公开总体上涉及人工智能的技术领域，并且具体地涉及一种处理张量数据的方法和装置。

背景技术

相关技术中提出的基于感兴趣区域(Region of Interest，ROI)的目标检测方案中，对于每个输入张量(tensor)，可以通过例如区域推荐网络(Region Proposal Network，RPN)等生成大量的ROI，例如，可以是数千个、数万个、数十万个、甚至更多的ROI。

然后，针对输入张量中与所生成的各个ROI相对应的各个局部张量，可以执行诸如排序、非极大值抑制、插值和对齐等各种处理，并且进一步地，可以通过例如基于区域的卷积神经网络(Region-based Convolution Neural Network，RCNN)等，获得针对输入张量的检测结果。

发明内容

根据本公开的一个方面，提供了一种处理张量数据的方法，该方法可以包括：存储多个指令序列，每个指令序列用于处理与该指令序列相对应的数量的张量；针对输入张量的每个感兴趣区域，从所述输入张量中确定与该感兴趣区域对应的局部张量；以及针对所获取的每第一数量的局部张量，执行所述多个指令序列中的用于处理所述第一数量的张量的第一指令序列。

根据本公开的另一个方面，还提供了一种处理张量数据的装置，该装置可以包括：指令存储模块，被配置为存储多个指令序列，每个指令序列用于处理与该指令序列相对应的数量的张量；张量获取模块，被配置为针对输入张量的每个感兴趣区域，从所述输入张量中获取与该感兴趣区域对应的局部张量；以及第一指令执行模块，被配置为针对通过所述张量获取模块所获取的每第一数量的局部张量，执行所述多个指令序列中的用于处理所述第一数量的张量的第一指令序列。

根据本公开的另一个方面，还提供了一种计算机可读存储介质，在其上存储有计算机程序，所述计算机程序用于执行上述方法。

根据本公开的另一个方面，还提供了一种电子设备，该电子设备可以包括处理器和用于存储所述处理器可执行的指令的存储器，所述处理器用于从所述存储器中读取所述指令，并执行所述指令以实现上述方法。

通过根据本公开的实施例的方法、装置和电子设备，能够减少在张量数据的处理过程中的数据传输量，提升张量数据的处理速度。

附图说明

通过结合附图对本公开实施例进行更详细的描述，本公开的上述以及其他目的、特征和优势将变得更加明显。附图用来提供对本公开实施例的进一步理解，并且构成说明书的一部分，与本公开实施例一起用于解释本公开，并不构成对本公开的限制。在附图中，相同的参考标号通常代表相同部件或步骤。

图1示出用于处理张量数据的示例装置。

图2示出根据本公开的实施例的用于处理张量数据的示例方法。

图3示出根据本公开的实施例的局部张量和ROI的示例。

图4示出根据本公开的实施例的处理过程的示例。

图5示出针对张量要执行的运算示例。

图6示出针对图5的运算的过程的示例。

图7示出根据本公开的实施例的针对图5的运算的过程的示例。

图8示出根据本公开的实施例的用于处理张量数据的示例方法。

图9示出根据本公开的实施例的用于处理张量数据的示例方法。