[發(fā)明專利]矩陣的空間局部變換

在審

• 申請?zhí)枺?/td>: 201980074328.9
• 申請日: 2019-11-19
• 公開（公告）號: CN113039541A
• 公開（公告）日: 2021-06-25
• 發(fā)明（設(shè)計）人: 喬納森·亞歷山大·羅斯;托馬斯·霍金斯;格雷戈里·邁克爾·索爾森;馬特·博伊德
• 申請（專利權(quán)）人: 格羅克公司
• 主分類號: G06F17/15
• 分類號: G06F17/15
• 代理公司: 北京集佳知識產(chǎn)權(quán)代理有限公司 11227
• 代理人: 唐京橋;何方

1.一種方法，包括：

接收與核進行卷積的輸入張量；

將所述輸入張量劃分成一個或更多個圖塊，每個圖塊的尺寸等于所述核的尺寸；以及

將所述一個或更多個圖塊中的值展平成向量，以生成展平的輸入流。

2.根據(jù)權(quán)利要求1所述的方法，還包括：

使用填充值填充所述輸入張量，使得使用所述核對所述輸入張量進行卷積的輸出的尺寸與所述輸入張量的尺寸相同。

3.根據(jù)權(quán)利要求1所述的方法，還包括：

使用填充值填充所述輸入張量，使得所述輸入張量的每個維度的尺寸是所述核的相應(yīng)維度的整數(shù)倍。

4.根據(jù)權(quán)利要求1所述的方法，還包括：

使用以下填充值填充所述輸入張量的每個維度的尾邊緣，所述填充值的寬度等于所述核在相應(yīng)維度中的尺寸。

5.根據(jù)權(quán)利要求4所述的方法，其中，所述填充值是零值。

6.根據(jù)權(quán)利要求1所述的方法，其中，所述核的第一側(cè)和第二側(cè)是垂直的，并且所述第一側(cè)的長度與所述第二側(cè)的長度不同，所述方法還包括：以與所述核跨越所述輸入張量的方向?qū)?zhǔn)的順序劃分所述輸入張量。

7.根據(jù)權(quán)利要求1所述的方法，其中，所述核的第一側(cè)和第二側(cè)是垂直的，并且所述第一側(cè)的長度與所述第二側(cè)的長度不同，所述方法還包括：以與所述核跨越所述輸入張量的方向正交的順序劃分所述輸入張量。

8.根據(jù)權(quán)利要求1所述的方法，其中，展平所述一個或更多個圖塊中的值還包括：對于所述輸入張量的一個或更多個圖塊中的每個圖塊，

以定義的順序訪問所述圖塊的值；

根據(jù)所述定義的順序?qū)⑺鲋挡贾迷谙蛄恐校灰约?/p>

以平行布置來布置與所述一個或更多個圖塊中的每個圖塊對應(yīng)的一個或更多個向量，以生成所述展平的輸入流。

9.根據(jù)權(quán)利要求1所述的方法，其中，所述定義的順序是行優(yōu)先順序、列優(yōu)先順序和通道優(yōu)先順序中至少之一，其中，所述通道優(yōu)先順序首先沿著與三維(3D)圖塊的深度對應(yīng)的軸訪問所述3D圖塊中的元素，然后沿著與所述3D圖塊的寬度和高度對應(yīng)的軸訪問所述3D圖塊中的元素。

10.根據(jù)權(quán)利要求1所述的方法，其中，所述展平的輸入流被存儲在緩存器中，所述緩存器由硬件加速處理器讀取以執(zhí)行與1)通過控制模式選擇的所述展平的輸入流中的值以及2)所述核的擴展的乘法加法運算，以生成卷積運算的輸出，而無需將所述輸入張量的值多次加載到所述緩存器中。

11.根據(jù)權(quán)利要求1所述的方法，其中，所述輸入張量具有多個通道，其中，所述核具有多個過濾器，并且其中，所述輸入通道與每個核過濾器卷積以生成具有多個輸出通道的輸出。

12.一種系統(tǒng)，包括：

處理器，所述處理器包括輸入流生成器，所述輸入流生成器被配置成：

接收與核進行卷積的輸入張量；

將所述輸入張量劃分成一個或更多個圖塊，每個圖塊的尺寸等于所述核的尺寸；以及

將所述一個或更多個圖塊中的值展平成向量，以生成所述展平的輸入流。

13.根據(jù)權(quán)利要求12所述的系統(tǒng)，其中，所述輸入流生成器還被配置成：

使用填充值填充所述輸入張量，使得使用所述核對所述輸入張量進行卷積的輸出的尺寸與所述輸入張量的尺寸相同。

14.根據(jù)權(quán)利要求12所述的系統(tǒng)，其中，所述輸入流生成器還被配置成：

使用填充值填充所述輸入張量，使得所述輸入張量的每個維度的尺寸是所述核的相應(yīng)維度的整數(shù)倍。