torch.nn.Unfold()详细解释
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 blo..
原创
2021-10-07 18:26:00 ·
4935 阅读 ·
3 评论