f-dangel / unfoldNd

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

Add documentation #3

Closed f-dangel closed 3 years ago

f-dangel commented 3 years ago

After #2, the API needs to be documented. I think it's best to explain what the function does in one line, then reference the documentation of torch.nn.Unfold to prevent duplication.