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