search

f-dangel / unfoldNd

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

Write example.py #8

Closed f-dangel closed 3 years ago

f-dangel commented 3 years ago

Currently, this is a dummy file. A simple comparison with torch.nn.Unfold should be sufficient.