TRANSPOSE OPERATIONS USING PROCESSING ELEMENT ARRAY

Provided are systems and methods for transposing a tensor using processing element array operations. In some cases, it may be necessary to transpose elements of a tensor to perform a matrix operation. The tensor may be decomposed into blocks of data elements having dimensions consistent with the dimensions of a systolic array. An identity multiplication may be performed on each block of data elements loaded into a systolic array and the multiplication products summed in column partitions of a results buffer. The data elements in the column partitions of results buffer can then be mapped to row partitions of a buffer memory for further processing. 本发明提供了用于使用处理元件阵列运算来转置张量的系统和方法。在一些情况下，可能需要转置张量的元素以执行矩阵运算。可以将所述张量分解成维度与脉动阵列的维度一致的数据元素块。可以对加载到脉动阵列中的每个数据元素块执行恒等乘法，并且在结果缓冲器的列分区中对乘积求和。然后，可以将结果缓冲器的所述列分区中的所述数据元素映射到缓冲存储器的行分区以供进一步处理。