Sha256: 412f76dbdca2c8583a129bfb8f69c221378f015d1d3b993dde85e8f20fd1692c
Contents?: true
Size: 502 Bytes
Versions: 54
Compression:
Stored size: 502 Bytes
Contents
module Torch
module NN
class Fold < Module
def initialize(output_size, kernel_size, dilation: 1, padding: 0, stride: 1)
super()
@output_size = output_size
@kernel_size = kernel_size
@dilation = dilation
@padding = padding
@stride = stride
end
def forward(input)
F.fold(input, @output_size, @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