[發明專利]用于計算的方法、計算設備和計算機可讀存儲介質

有效

• 申請號: 202110267756.2
• 申請日: 2021-03-12
• 公開（公告）號: CN112862725B
• 公開（公告）日: 2023-10-27
• 發明（設計）人: 請求不公布姓名
• 申請（專利權）人: 上海壁仞智能科技有限公司
• 主分類號: G06T5/00
• 分類號: G06T5/00
• 代理公司: 北京市金杜律師事務所 11256
• 代理人: 黃倩

本公開涉及一種用于計算的方法、計算設備和計算機可讀存儲介質。該方法包括在高速緩存中確定第一像素塊，第一像素塊由2m行*2n列的像素矩陣組成并且包括原始像素數據以及與原始像素數據有關的像素數據。然后，從高速緩存中讀取第一像素塊。與原始像素數據有關的像素數據中的至少部分像素數據用于與原始像素數據有關的填充，并且原始像素數據包括2m行*2n列的像素矩陣中的第m+1行到第2m行中的第n+1列到第2n列的像素數據。由此，能夠實現在從高速緩存中讀取數據時，一次讀取在反向傳播中在對原始像素數據進行插零操作和填充操作后所需獲得的像素數據，從而降低了系統功耗，提高了系統性能。

技術領域

本公開的實施例總體上涉及人工智能領域，更具體地涉及一種用于計算的方法、計算設備和計算機可讀存儲介質。

背景技術

填充像素數據經常出現在基于人工智能的圖像處理或深度神經網絡中。例如在深度神經網絡中，在對像素數據進行處理前，通常會對原始像素數據進行填充(Padding)操作。在一個示例中，可以通過在將被處理的原始像素數據的邊界周圍添加額外的像素，實現對原始圖像的填充。

在反向傳播(Back Propagration)的權重梯度計算中，通常需要使用例如在水平方向上進行填充和/或在垂直方向上進行填充來獲得經填充的像素數據。

發明內容

本公開提供了一種用于計算的方法、計算設備和計算機可讀存儲介質，能夠提供一種用于計算的高速緩存實現。

根據本公開的第一方面，提供了一種用于計算的方法。該方法包括：在高速緩存中確定第一像素塊，第一像素塊由2m行*2n列的像素矩陣組成并且包括原始像素數據以及與原始像素數據有關的像素數據，m和n為正整數；以及從高速緩存中讀取第一像素塊，其中與原始像素數據有關的像素數據中的至少部分像素數據用于與原始像素數據有關的填充，并且原始像素數據包括2m行*2n列的像素矩陣中的第m+1行到第2m行中的第n+1列到第2n列的像素數據。

根據本發明的第二方面，還提供了一種計算設備，該設備包括：至少一個處理單元；以及至少一個存儲器，至少一個存儲器被耦合到至少一個處理單元并且存儲用于由至少一個處理單元執行的指令，指令當由至少一個處理單元執行時，使得設備執行根據第一方面的方法的步驟。

根據本公開的第三方面，還提供了一種計算機可讀存儲介質。該計算機可讀存儲介質上存儲有計算機程序，計算機程序被機器執行時執行本公開的第一方面的方法。

在一些實施例中，該方法還包括基于與原始像素數據有關的填充配置所指示的在垂直方向上的坐標，從2m行*2n列的像素矩陣的2m行中確定p行，以得到p行*2n列的像素矩陣；基于填充配置所指示的在水平方向上的坐標，從所確定的p行*2n列的像素矩陣的2n列中確定q列，以得到p行*q列的像素矩陣，p和q為正整數；對p行*q列的像素矩陣進行插零操作，以將p行*q列的像素矩陣擴展為2p行*2q列的像素矩陣。

在一些實施例中，2m行為4行，從2m行*2n列的像素矩陣的2m行中確定p行包括：響應于填充配置所指示的在垂直方向上的坐標為+3或+2時，從4行*2n列的像素矩陣的4行中選擇第2行和第3行，以得到2行*2n列的像素矩陣；以及響應于填充配置所指示的在垂直方向上的坐標為+1或0時，從4行*2n列的像素矩陣的4行中選擇第3行和第4行，以得到2行*2n列的像素矩陣。

在一些實施例中，2n列為8列，從所確定的p行*2n列的像素矩陣的2n列中確定q列包括：響應于填充配置所指示的在水平方向上的坐標為0或+1時，從2行*2n列的像素矩陣的8列中選擇第5列至第8列，以得到2行*4列的像素矩陣；并且響應于填充配置所指示的在水平方向上的坐標為+2或+3時，從2行*2n列的像素矩陣的8列中選擇第4列至第7列，以得到2行*4列的像素矩陣。

在一些實施例中，對p行*q列的像素矩陣進行插零操作包括：對p行像素矩陣隔行插一行零；以及對q行像素矩陣隔列插一列零，以得到2p行*2q列的像素矩陣。