Extracts sliding local blocks from an batched input tensor.
nnf_unfold(input, kernel_size, dilation = 1, padding = 0, stride = 1)the input tensor
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 input tensors (batched image-like tensors) are supported.
More than one element of the unfolded tensor may refer to a single memory location. As a result, in-place operations (especially ones that are vectorized) may result in incorrect behavior. If you need to write to the tensor, please clone it first.