文件名称：butterfly:PyTorch中的蝴蝶矩阵乘法
文件大小：497KB
文件格式：ZIP
更新时间：2024-06-04 12:47:48
Python
随附论文的代码，其中和。 要求 python> = 3.6 火炬> = 1.8 麻木科学的 用法 2020-08-03：蝴蝶C ++ / CUDA代码的新接口位于csrc和torch_butterfly 。 它在tests/test_butterfly.py进行了tests/test_butterfly.py （还显示了示例用法）。 安装： python setup.py install 也就是说，使用此根目录中的setup.py文件。 文件torch_butterfly/special.py显示了如何构造具有FFT，逆FFT，循环矩阵乘法，Hadamard变换和带有圆形填充的torch.nn.Conv1d的蝶形矩阵。 tests/test_special.py中的tests/test_special.py表明，这些蝶形矩阵完全执行了这些操作。 旧介面 注意：此接口正在被重写。 仅