#池化层
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通道
22-池化层 动手深度学习
最新推荐文章于 2024-08-03 18:33:38 发布