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