Sha256: e33038a5ee13839ff481421619e53b7b9fed3cecb9bd1f8d44b79e859465abe6
Contents?: true
Size: 605 Bytes
Versions: 47
Compression:
Stored size: 605 Bytes
Contents
module Torch
module NN
class MaxPoolNd < Module
def initialize(kernel_size, stride: nil, padding: 0, dilation: 1, return_indices: false, ceil_mode: false)
super()
@kernel_size = kernel_size
@stride = stride || kernel_size
@padding = padding
@dilation = dilation
@return_indices = return_indices
@ceil_mode = ceil_mode
end
def extra_inspect
s = "kernel_size: %{kernel_size}, stride: %{stride}, padding: %{padding}, dilation: %{dilation}, ceil_mode: %{ceil_mode}"
format(s, **dict)
end
end
end
end
Version data entries
47 entries across 47 versions & 1 rubygems