search

f-dangel / unfoldNd

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

Revert "Avoid repeating kernel" #13

Closed f-dangel closed 3 years ago

f-dangel commented 3 years ago

Reverts f-dangel/unfoldNd#12.

The benchmark results deteriorated both in terms of run time and memory consumption.