Sha256: 6231660ad7403ab92801ce5a6d68e2e898061babc3c4eb104622866948ea16a8
Contents?: true
Size: 584 Bytes
Versions: 54
Compression:
Stored size: 584 Bytes
Contents
module Torch
module NN
class LPPoolNd < Module
def initialize(norm_type, kernel_size, stride: nil, ceil_mode: false)
super()
@norm_type = norm_type
@kernel_size = kernel_size
@stride = stride
@ceil_mode = ceil_mode
end
def extra_inspect
format("norm_type: %{norm_type}, kernel_size: %{kernel_size}, stride: %{stride}, ceil_mode: %{ceil_mode}",
norm_type: @norm_type,
kernel_size: @kernel_size,
stride: @stride,
ceil_mode: @ceil_mode
)
end
end
end
end
Version data entries
54 entries across 54 versions & 1 rubygems