Sha256: ee1c319e03f43c429860c6f4bb2190fb3c62c0c628658dc4b4a97d782d0d392c
Contents?: true
Size: 471 Bytes
Versions: 54
Compression:
Stored size: 471 Bytes
Contents
module Torch
module NN
class MaxUnpool1d < MaxUnpoolNd
def initialize(kernel_size, stride: nil, padding: 0)
super()
@kernel_size = _single(kernel_size)
@stride = _single(stride || kernel_size)
@padding = _single(padding)
end
def forward(input, indices, output_size: nil)
F.max_unpool1d(input, indices, @kernel_size, stride: @stride, padding: @padding, output_size: output_size)
end
end
end
end
Version data entries
54 entries across 54 versions & 1 rubygems