torch.nn.Unfold(kernel_size, dilation=1, padding=0, stride=1)
功能:从一个批次的输入张量中提取出滑动的局部区域块。(Extracts sliding local blocks from a batched input tensor.)
参数:
- kernel_size (intortuple) – 滑块的大小
- stride (intortuple,optional) – 滑块的步长(the stride of the sliding blocks in the input