Combines an array of sliding local blocks into a large containing tensor.
nnf_fold(
input,
output_size,
kernel_size,
dilation = 1,
padding = 0,
stride = 1
)
the input tensor
the shape of the spatial dimensions of the output (i.e.,
output$sizes()[-c(1,2)]
)
the size of the sliding blocks
a parameter that controls the stride of elements within the neighborhood. Default: 1
implicit zero padding to be added on both sides of input. Default: 0
the stride of the sliding blocks in the input spatial dimensions. Default: 1
Currently, only 4-D output tensors (batched image-like tensors) are supported.