tensorplay.nn.functional.unfold

tensorplay.nn.functional.unfold(input, kernel_size, dilation=1, padding=0, stride=1)[source]

Extract sliding local blocks from a batched input tensor (torch F.unfold, i.e. aten::im2col). 4-D batched input is the common case; unbatched 3-D input is accepted like aten’s im2col.