Sha256: bde233a9add0bce6c651e18662f134e59b0ee0faf8c35cfca665fb98a5656b47
Contents?: true
Size: 599 Bytes
Versions: 54
Compression:
Stored size: 599 Bytes
Contents
module Torch
module NN
class AvgPool2d < AvgPoolNd
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
def forward(input)
F.avg_pool2d(input, @kernel_size, @stride, @padding, @ceil_mode, @count_include_pad, @divisor_override)
end
end
end
end
Version data entries
54 entries across 54 versions & 1 rubygems