f-dangel / unfoldNd

(N=1,2,3)-dimensional unfold (im2col) and fold (col2im) in PyTorch
MIT License
82 stars 6 forks source link

unfold for transpose convolution #16

Closed f-dangel closed 1 year ago

f-dangel commented 3 years ago

The input to a transpose convolution can also be unfolded. Then, the forward pass (with zero bias) is a matrix-matrix multiplication between unfolded input and matrix-view of the kernel.