# 池化层的正向传播 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.sjape[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
torch池化层基本原理代码
最新推荐文章于 2024-08-02 03:58:58 发布