22-池化层 动手深度学习

#池化层
import torch
from torch import nn
from d2l import torch as d2l
#实现池化层的正向传播
def pool2d(x,pool_size,mode='max'):
    p_h,p_w = pool_size
    y = torch.zeros((x.shape[0] - p_h + 1,x.shape[1] - p_w + 1))
    for i in range(y.shape[0]):
        for j in range(y.shape[1]):
            if mode == 'max':
                y[i,j] = x[i:i+p_h,j:j+p_w].max()#最大池化层
            elif mode == 'avg':
                y[i,j] = x[i:i+p_h,j:j+p_w].mean()#平均池化层
    return y

x = torch.tensor([[0.0,1.0,2.0],[3.0,4.0,5.0],[6.0,7.0,8.0]])
print(pool2d(x,(2,2)))
# tensor([[4., 5.],
#         [7., 8.]])
print(pool2d(x,(2,2),'avg'))
# tensor([[2., 3.],
#         [5., 6.]])

#填充和步幅
x = torch.arange(16,dtype=torch.float32).reshape((1,1,4,4))#通道1批量1 4*4矩阵
print(x)
# tensor([[[[ 0.,  1.,  2.,  3.],
#           [ 4.,  5.,  6.,  7.],
#           [ 8.,  9., 10., 11.],
#           [12., 13., 14., 15.]]]])
pool2d = nn.MaxPool2d(3)#3*3窗口 深度学习中的步幅与池化窗口的大小相同
print(pool2d(x))
# tensor([[[[10.]]]])

pool2d = nn.MaxPool2d(3,padding=1,stride=2)
print(pool2d(x))
# tensor([[[[ 5.,  7.],
#           [13., 15.]]]])
pool2d = nn.MaxPool2d((2,3),padding=(1,1),stride=(2,3))
print(pool2d(x))
# tensor([[[[ 1.,  3.],
#           [ 9., 11.],
#           [13., 15.]]]])

x = torch.cat((x,x+1),1)
pool2d = nn.MaxPool2d(3,padding=1,stride=2)
print(pool2d(x))#2通道