torch.nn.unfold && torch.nn.Fold

本文转载自查看原文 2021-04-18 10:43 389 pytorch

torch.nn.unfold

提取滑动窗口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)

View Code

免责声明！

本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。

猜您在找 F.unfold F.fold F.normalize() Pytorch: torch.nn 到底什么是TORCH.NN？ torch.nn.CrossEntropyLoss torch.nn.Parameter 详解 pytorch的torch.nn.CrossEntropyLoss() 关于torch.nn.Linear的笔记 torch.nn.Linear解释 torch.nn.BCELoss用法 Torch7学习笔记（二）nn Package