Sha256: b42182103496cf4cc914ede9fb1506e17d85e59a6849ad2073bb2f1b0491530b
Contents?: true
Size: 444 Bytes
Versions: 54
Compression:
Stored size: 444 Bytes
Contents
module Torch
module NN
class Unfold < Module
def initialize(kernel_size, dilation: 1, padding: 0, stride: 1)
super()
@kernel_size = kernel_size
@dilation = dilation
@padding = padding
@stride = stride
end
def forward(input)
F.unfold(input, @kernel_size, dilation: @dilation, padding: @padding, stride: @stride)
end
# TODO add extra_inspect
end
end
end
Version data entries
54 entries across 54 versions & 1 rubygems