pytorch-complex icon indicating copy to clipboard operation
pytorch-complex copied to clipboard

pytorch-complex

Install it using pip:

pip install pytorch-complex

Usage: Similar to PyTorch. For using the Complex features of this library, just change the regular torch imports with torchcomplex imports. For example: import torchcomplex.nn as nn instead of import torch.nn as nn Then, simply nn.Conv2d for both torch and torchcomplex, for 2D Convolution

Credits

If you like this repository, please click on Star!

If you use this package or benift from the codes of this repo, please cite the following in your publications:

Soumick Chatterjee, Pavan Tummala, Oliver Speck, Andreas Nürnberger: Complex Network for Complex Problems: A comparative study of CNN and Complex-valued CNN (IEEE IPAS, Dec 2023)

BibTeX entry:

@inproceedings{chatterjee2022complex,
  title={Complex Network for Complex Problems: A comparative study of CNN and Complex-valued CNN},
  author={Chatterjee, Soumick and Tummala, Pavan and Speck, Oliver and N{\"u}rnberger, Andreas},
  booktitle={2022 IEEE 5th International Conference on Image Processing Applications and Systems (IPAS)},
  pages={1--5},
  year={2022},
  organization={IEEE}
}

Thank you so much for your support.