[ADD] input unfolding for transpose convolution

f-dangel / unfoldNd

(N=1,2,3)-dimensional unfold (im2col) and fold (col2im) in PyTorch

MIT License

82 stars 6 forks source link

[ADD] input unfolding for transpose convolution #27

Closed f-dangel closed 1 year ago

f-dangel commented 1 year ago

This PR adds the equivalent of torch.nn.Unfold (unfolding the input of a convolution torch.nn.ConvNd) for transpose convolution (unfolding the input of a transpose convolution torch.nn.ConvTransposeNd).

Resolves #16.

f-dangel commented 1 year ago

Remaining TODOs:

[x] Update docstring of _make_weight
[x] Look at untested lines, maybe add additional tests
[x] Update README
[x] Prepare an example
[x] Handle changed signature of _output_padding in different PyTorch releases