Class: Qoa::Layers::PoolingLayer

Inherits:
Layer
  • Object
show all
Defined in:
lib/qoa/layers/pooling_layer.rb

Instance Attribute Summary collapse

Attributes inherited from Layer

#input_size, #output_size, #weights

Instance Method Summary collapse

Methods inherited from Layer

#random_matrix

Constructor Details

#initialize(input_size, output_size, pool_size, stride = 1) ⇒ PoolingLayer



6
7
8
9
10
# File 'lib/qoa/layers/pooling_layer.rb', line 6

def initialize(input_size, output_size, pool_size, stride = 1)
  super(input_size, output_size)
  @pool_size = pool_size
  @stride = stride
end

Instance Attribute Details

#pool_sizeObject (readonly)

Returns the value of attribute pool_size.



4
5
6
# File 'lib/qoa/layers/pooling_layer.rb', line 4

def pool_size
  @pool_size
end

#strideObject (readonly)

Returns the value of attribute stride.



4
5
6
# File 'lib/qoa/layers/pooling_layer.rb', line 4

def stride
  @stride
end