池化-MaxPool2d,AvgPool2d

最新推荐文章于 2024-09-24 14:49:05 发布

周小天..

最新推荐文章于 2024-09-24 14:49:05 发布

阅读量687

点赞数

分类专栏：图像处理文章标签：深度学习

本文链接：https://blog.csdn.net/xingghaoyuxitong/article/details/112941239

版权

图像处理专栏收录该内容

53 篇文章 3 订阅

订阅专栏

#%%
import torch
from torch import nn
#%% 最大值池化
arry = torch.arange(16,dtype=torch.float32).view([1,1,4,4])
print(arry)

maxp = nn.MaxPool2d(kernel_size=2)        #kernel_size=2,stride=kernel_size
Avgp = nn.AvgPool2d(kernel_size=2)        #kernel_size=2,stride=kernel_size
    
print(maxp(arry))
print(Avgp(arry))


tensor([[[[ 0.,  1.,  2.,  3.],
          [ 4.,  5.,  6.,  7.],
          [ 8.,  9., 10., 11.],
          [12., 13., 14., 15.]]]])
最大值池化： tensor([[[[ 5.,  7.],
          [13., 15.]]]])
均值池化  ： tensor([[[[ 2.5000,  4.5000],
          [10.5000, 12.5000]]]])