- torch.nn.functional.unfold(input, kernel_size, dilation=1, padding=0, stride=1)¶
Extracts sliding local blocks from a batched input tensor.
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.
- Return type: