torch.unfold

torch.unfold"/>

torch.unfold

项目场景：

针对torch中的unfold（只卷不积）：

问题描述

提示：这里描述项目中遇到的问题：
原始参数为
下面展示一些 内联代码片。

tensor([[[ 0.,  1.,  2.],[ 3.,  4.,  5.],[ 6.,  7.,  8.]],[[ 9., 10., 11.],[12., 13., 14.],[15., 16., 17.]],[[18., 19., 20.],[21., 22., 23.],[24., 25., 26.]]])

例如：输入数据为3×3×3的矩阵，使用该函数
x1 = f.unfold(x, kernel_size=3, dilation=1 , stride=1)

tensor([[ 0.],[ 1.],[ 2.],[ 3.],[ 4.],[ 5.],[ 6.],[ 7.],[ 8.],[ 9.],[10.],[11.],[12.],[13.],[14.],[15.],[16.],[17.],[18.],[19.],[20.],[21.],[22.],[23.],[24.],[25.],[26.]])}

若将参数替换为kernel=2

tensor([[ 0.,  1.,  3.,  4.],[ 1.,  2.,  4.,  5.],[ 3.,  4.,  6.,  7.],[ 4.,  5.,  7.,  8.],[ 9., 10., 12., 13.],[10., 11., 13., 14.],[12., 13., 15., 16.],[13., 14., 16., 17.],[18., 19., 21., 22.],[19., 20., 22., 23.],[21., 22., 24., 25.],[22., 23., 25., 26.]])}

再将diat=2

tensor([[ 0.],[ 2.],[ 6.],[ 8.],[ 9.],[11.],[15.],[17.],[18.],[20.],[24.],[26.]])```

类似于空洞的概念，如果将输入换成3×5×5就换过来了。

tensor([[[ 0.,  1.,  2.,  3.,  4.],[ 5.,  6.,  7.,  8.,  9.],[10., 11., 12., 13., 14.],[15., 16., 17., 18., 19.],[20., 21., 22., 23., 24.]],[[25., 26., 27., 28., 29.],[30., 31., 32., 33., 34.],[35., 36., 37., 38., 39.],[40., 41., 42., 43., 44.],[45., 46., 47., 48., 49.]],[[50., 51., 52., 53., 54.],[55., 56., 57., 58., 59.],[60., 61., 62., 63., 64.],[65., 66., 67., 68., 69.],[70., 71., 72., 73., 74.]]])

以上为输如，将size设为3，看输出

tensor([[ 0.,  1.,  2.,  5.,  6.,  7., 10., 11., 12.],[ 1.,  2.,  3.,  6.,  7.,  8., 11., 12., 13.],[ 2.,  3.,  4.,  7.,  8.,  9., 12., 13., 14.],[ 5.,  6.,  7., 10., 11., 12., 15., 16., 17.],[ 6.,  7.,  8., 11., 12., 13., 16., 17., 18.],[ 7.,  8.,  9., 12., 13., 14., 17., 18., 19.],[10., 11., 12., 15., 16., 17., 20., 21., 22.],[11., 12., 13., 16., 17., 18., 21., 22., 23.],[12., 13., 14., 17., 18., 19., 22., 23., 24.],[25., 26., 27., 30., 31., 32., 35., 36., 37.],[26., 27., 28., 31., 32., 33., 36., 37., 38.],[27., 28., 29., 32., 33., 34., 37., 38., 39.],[30., 31., 32., 35., 36., 37., 40., 41., 42.],[31., 32., 33., 36., 37., 38., 41., 42., 43.],[32., 33., 34., 37., 38., 39., 42., 43., 44.],[35., 36., 37., 40., 41., 42., 45., 46., 47.],[36., 37., 38., 41., 42., 43., 46., 47., 48.],[37., 38., 39., 42., 43., 44., 47., 48., 49.],[50., 51., 52., 55., 56., 57., 60., 61., 62.],[51., 52., 53., 56., 57., 58., 61., 62., 63.],[52., 53., 54., 57., 58., 59., 62., 63., 64.],[55., 56., 57., 60., 61., 62., 65., 66., 67.],[56., 57., 58., 61., 62., 63., 66., 67., 68.],[57., 58., 59., 62., 63., 64., 67., 68., 69.],[60., 61., 62., 65., 66., 67., 70., 71., 72.],[61., 62., 63., 66., 67., 68., 71., 72., 73.],[62., 63., 64., 67., 68., 69., 72., 73., 74.]])

将dia=2时

tensor([[ 0.],[ 2.],[ 4.],[10.],[12.],[14.],[20.],[22.],[24.],[25.],[27.],[29.],[35.],[37.],[39.],[45.],[47.],[49.],[50.],[52.],[54.],[60.],[62.],[64.],[70.],[72.],[74.]])

问题描述

终于理解了，是按照列的规矩垂直展平！！！！！