Sha256: 2ec67df40be507a6f79b5138e9459cdcb01bf18bbf2ac704831471133aea4472
Contents?: true
Size: 599 Bytes
Versions: 54
Compression:
Stored size: 599 Bytes
Contents
module Torch
module NN
class AvgPool3d < 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_pool3d(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