Unfold関数について - Qiita

#Conv2D演算 2次元畳み込みの演算を考えた場合、入力$(Batch,H,W,C_{in})$、出力$(Batch,H,W,C_{out})$、カーネルサイズ$(3,3)$、畳み込みの重み$W=(3,3,C_{in},C_{out})$とすれば、 実質的にConv2D演算は $matmul(x,W)=matmul((Batch,HW,9C_{in}), (9C_{in}, C_{out}))=(Batch,HW,C_{out})$ というmatmulの行列演算を考えることに等しいです。 ここで$c=matmul(a,b)$という行列演算において$a=(i,j,k,m),b=(m,n)$ならば$c=(i,j,k,n)$です。 一方で入力$(Batch,H,W,C_{in})$に対して x[:,0]=input[:,0:H-2,0:W-2,:] \\ x[:,1]=input[:,0:H-