Class: Chainer::Functions::Pooling::Pooling2D

Inherits:
Chainer::Function show all
Defined in:
lib/chainer/functions/pooling/pooling_2d.rb

Overview

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

Direct Known Subclasses

AveragePooling2D, MaxPooling2D

Instance Attribute Summary

Attributes inherited from Chainer::Function

#inputs, #output_data, #outputs, #rank, #retain_after_backward

Instance Method Summary collapse

Methods inherited from Chainer::Function

#backward, #call, #forward, #forward_cpu, #retain_inputs, #retain_outputs

Constructor Details

#initialize(ksize, stride: nil, pad: 0, cover_all: true) ⇒ Pooling2D

Returns a new instance of Pooling2D.



6
7
8
9
10
11
12
13
14
15
16
 
# File 'lib/chainer/functions/pooling/pooling_2d.rb', line 6

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