提取滑動窗口patches
1 inputs = torch.randn(1, 2, 4, 4) 2 print(inputs.size()) 3 print(inputs) 4 unfold = torch.nn.Unfold(kernel_size=(2, 2), stride=2) 5 patches = unfold(inputs) 6 print(patches.size()) 7 print(patches)
torch.nn.unfold && torch.nn.Fold
免責聲明！
本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。