Method: Torch::NN::AvgPool1d#initialize

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

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

Returns a new instance of AvgPool1d.



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

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