池化层【马赛克】

最大池化,也叫下采样。

公式:

参数:

celling的意思:

例子:

尺寸:

code:

import torchimport torchvision.datasetsfrom tensorboardX import SummaryWriterfrom torch import nnfrom torch.nn import MaxPool2dfrom torch.utils.data import DataLoaderdataset = 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  outputtudui = 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()

相关推荐

  1. 什么是

    2024-06-09 16:42:03       26 阅读
  2. NLP(7)--Embedding、、丢弃

    2024-06-09 16:42:03       12 阅读
  3. 【深度学习基础】

    2024-06-09 16:42:03       2 阅读
  4. pytorch学习5-最大的使用

    2024-06-09 16:42:03       39 阅读

最近更新

  1. (一)PHP 变量

    2024-06-09 16:42:03       0 阅读
  2. docker

    docker

    2024-06-09 16:42:03      0 阅读
  3. php环境变量$_ENV详解

    2024-06-09 16:42:03       0 阅读

热门阅读

  1. 简单记录个python国内镜像源

    2024-06-09 16:42:03       4 阅读
  2. Spring Boot配置MySQL数据库连接数

    2024-06-09 16:42:03       5 阅读
  3. vue路由缓存

    2024-06-09 16:42:03       5 阅读
  4. 力扣第185题:部门工资前三高的员工

    2024-06-09 16:42:03       3 阅读
  5. 手机UI设计中的按钮状态包含哪几种

    2024-06-09 16:42:03       5 阅读
  6. 等级保护与网络安全:构建信息安全的坚实防线

    2024-06-09 16:42:03       4 阅读