import torch
x = torch.arange(8*12).view(1,1,8,12)
m=x.unfold(2, 4, 4)
n = m.unfold(3, 4, 4)
輸入
第一次切,切高度維度,但是切完做了轉置 ,得到(1,1,2,12,4)
切寬度? 得到 張量維度1 1 2 3 4 4? 最后得到 維度表示:批次 通道 高度切塊數 寬度切塊數 高度 寬度
patches.contiguous().view(B, -1, 4*4)其中contiguous()不用管,不改變張量形狀.view(B, -1, 4*4),把張量第一維度是 B,第二維度-1代表自動計算大小,其實就是總共切塊的數目,最后一維度代表一個塊的大小