guoyouwei88 / cplxmodule

A lightweight neural network extension for pytorch that handles complex valued inputs respecting complex algebraic operations.

CplxModule

A lightweight extension for pytorch.nn that adds layers and activations, which respect algebraic operations over the field of complex numbers.

The implementation is based on the ICLR 2018 parer on Deep Complex Networks [1]_ and borrows ideas from their implementation.

Installation

Just run to imnstall with pip

pip install --upgrade git+https://github.com/ivannz/cplxmodule.git

or

python setup.py install

to install from thie root of thie repo.

Example

Basically the module is designed in such a way as to be ready for plugging into the existing torch.nn sequential models.

import torch
import torch.nn

pass

References

.. [1] Trabelsi, C., Bilaniuk, O., Zhang, Y., Serdyuk, D., Subramanian, S., Santos, J. F., ... & Pal, C. J. (2017). Deep complex networks. arXiv preprint arXiv:1705.09792