设输入张量:[b, c, h, w]
unfold参数:kernel_size=step
则输出的张量:[b0, n, l]
b0=b
n=c*kernel_size*kernel_size
l=⌊h/kernel_size⌋ *⌊w/kernel_size⌋
设输入张量:[b, c, h, w]
unfold参数:kernel_size=step
则输出的张量:[b0, n, l]
b0=b
n=c*kernel_size*kernel_size
l=⌊h/kernel_size⌋ *⌊w/kernel_size⌋