class Chainer::Functions::Pooling::Pooling2D

Base class of pooling function over a set of 2d planes

Attributes

cover_all[R]
kh[R]
kw[R]
ph[R]
pw[R]
sx[R]
sy[R]

Public Class Methods

new(ksize, stride: nil, pad: 0, cover_all: true) click to toggle source
# File lib/chainer/functions/pooling/pooling_2d.rb, line 8
def initialize(ksize, stride: nil, pad: 0, cover_all: true)
  if stride.nil?
    stride = ksize
  end

  @kh, @kw = ksize.is_a?(::Array) ? ksize : [ksize, ksize]
  @sy, @sx = stride.is_a?(::Array) ? stride : [stride, stride]
  @ph, @pw = pad.is_a?(::Array) ? pad: [pad, pad]

  @cover_all = cover_all
end