Method: Torch::NN::AvgPool3d#initialize

Defined in:
lib/torch/nn/avg_pool3d.rb

#initialize(kernel_size, stride: nil, padding: 0, ceil_mode: false, count_include_pad: true, divisor_override: nil) ⇒ AvgPool3d

Returns a new instance of AvgPool3d.



4
5
6
7
8
9
10
11
12
 
# File 'lib/torch/nn/avg_pool3d.rb', line 4

def initialize(kernel_size, stride: nil, padding: 0, ceil_mode: false, count_include_pad: true, divisor_override: nil)
  super()
  @kernel_size = kernel_size
  @stride = stride || kernel_size
  @padding = padding
  @ceil_mode = ceil_mode
  @count_include_pad = count_include_pad
  @divisor_override = divisor_override
end