Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.
MIT License
tensor4 - pytorch to C++ convertor using lightweight templated tensor library
This is a pytorch implementation of k-means clustering algorithm