您好,登錄后才能下訂單哦!
本篇內容介紹了“torch.nn.AvgPool2d的使用方法是什么”的有關知識,在實際案例的操作過程中,不少人都會遇到這樣的困境,接下來就讓小編帶領大家學習一下如何處理這些情況吧!希望大家仔細閱讀,能夠學有所成!
代碼實驗展示:
C:\Users\chenxuqi>conda activate ssd4pytorch2_2_0(ssd4pytorch2_2_0) C:\Users\chenxuqi>python Python 3.7.7 (default, May 6 2020, 11:45:54) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32 Type "help", "copyright", "credits" or "license" for more information.>>> import torch>>> torch.manual_seed(seed=20200910)<torch._C.Generator object at 0x000002701915D330>>>> data_in = torch.randn(3,5,100,100)>>> data_in.shape torch.Size([3, 5, 100, 100])>>>>>> from torch import nn>>> model = nn.AvgPool2d(4, stride=2)>>> data_out = model(data_in)>>> data_in.shape torch.Size([3, 5, 100, 100])>>> data_out.shape torch.Size([3, 5, 49, 49])>>> data_out = torch.nn.functional.avg_pool2d(data_in, kernel_size=4, stride=8)>>> data_out.shape torch.Size([3, 5, 13, 13])>>>>>> data_out = torch.nn.functional.avg_pool2d(data_in, kernel_size=4, stride=2)>>> data_out.shape torch.Size([3, 5, 49, 49])>>>>>>>>>>>>>>>
“torch.nn.AvgPool2d的使用方法是什么”的內容就介紹到這里了,感謝大家的閱讀。如果想了解更多行業相關的知識可以關注億速云網站,小編將為大家輸出更多高質量的實用文章!
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。