zoukankan      html  css  js  c++  java
  • pytorch卷积的输入输出以及计算公式

    1、nn.Conv2d
    class torch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True)
    二维卷积层, 输入的尺度是(N, C_in, H, W),输出尺度(N, C_out, H_out, W_out)。

    [out = (input + 2 * padding - kernel\_size) / stride + 1 ]

    2、nn.ConvTranspose2d
    class torch.nn.ConvTranspose2d(in_channels, out_channels, kernel_size, stride=1, padding=0, output_padding=0, groups=1, bias=True, dilation=1)

    [out = (input - 1) * stride - 2 * padding + kernel\_size ]

    3、nn.MaxPool2d
    class torch.nn.MaxPool2d(kernel_size, stride=None, padding=0, dilation=1, return_indices=False, ceil_mode=False)

    [out = (input + 2 * padding - kernel\_size) / stride + 1 ]

    4、nn.Conv3d
    class torch.nn.Conv3d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True)

    [out = (imput + 2 * padding - kernel\_size) / stride + 1 ]

    5、nn.ConvTranspose3d
    class torch.nn.ConvTranspose3d(in_channels, out_channels, kernel_size, stride=1, padding=0, output_padding=0, groups=1, bias=True)

    [out = (input - 1) * stride - 2 * padding + kernel\_size ]

  • 相关阅读:
    TCP/IP:IP的分片与重装
    TCP/IP:IP选项处理
    TCP/IP:IP多播选路
    TCP/IP:IGMP Internet组管理协议
    TCP/IP 插口选项
    TCP/IP: 插口I/O
    TCP/IP 插口层
    leetcode136只出现一次的数字
    leetcode268缺失数字
    letecode242有效字母的异位词
  • 原文地址:https://www.cnblogs.com/zyr001/p/14539849.html
Copyright © 2011-2022 走看看