池化层【马赛克】

最大池化,也叫下采样。

公式:

参数:

celling的意思:

例子:

尺寸:

code:

复制代码
import torchimport torchvision.datasetsfrom tensorboardX import SummaryWriterfrom torch import nnfrom torch.nn import MaxPool2dfrom torch.utils.data import DataLoader​dataset = torchvision.datasets.CIFAR10("data",train=False,download=True,transform=torchvision.transforms.ToTensor())dataloader = DataLoader(dataset,batch_size=64)​input = torch.tensor([[1,2,0,3,1],                      [0,1,2,3,1],                      [1,2,1,0,0],                      [5,2,3,1,1],                      [2,1,0,1,1]],dtype=torch.float32)​input = torch.reshape(input,(-1,1,5,5))print(input.shape)​class Tudui(nn.Module):    def __init__(self):        super(Tudui,self).__init__()        self.maxpool1 = MaxPool2d(kernel_size=3,ceil_mode = True)    def forward(self,input):        output = self.maxpool1(input)        return  output​tudui = Tudui()output = tudui(input)print(output)​writer = SummaryWriter("logs_maxpool")step = 0for data in dataloader:    imgs,targets = data    writer.add_images("input",imgs,step)    output=tudui(imgs)    writer.add_images("output",output,step)    step = step+1writer.close()
相关推荐
刘逸潇20056 分钟前
FastAPI(二)——请求与响应
网络·python·fastapi
java1234_小锋8 分钟前
TensorFlow2 Python深度学习 - 循环神经网络(SimpleRNN)示例
python·深度学习·tensorflow·tensorflow2
java1234_小锋10 分钟前
TensorFlow2 Python深度学习 - 通俗理解池化层,卷积层以及全连接层
python·深度学习·tensorflow·tensorflow2
fsnine21 分钟前
Python图形化界面——pyqt5教程
开发语言·python·qt
扶尔魔ocy1 小时前
python程序打包成win的exe应用(以OCR应用为例)
python·ocr·中文识别
Psycho_MrZhang1 小时前
自定义层和读写文件
pytorch·python·深度学习
似水流年 光阴已逝2 小时前
从Excel姓名匹配案例学Python:由点及面的系统化学习指南
开发语言·python·excel
这里有鱼汤2 小时前
量化的困局:当所有人都在跑同一个因子时,我们还能赚谁的钱?
后端·python
新子y2 小时前
【小白笔记】strip的含义
笔记·python
Q_Q19632884752 小时前
python+vue的在线租房 房屋租赁系统
开发语言·vue.js·spring boot·python·django·flask·node.js