boeddeker / pytorch-complex

complex tensor plugin for pytorch (Work in progress)

PyTorch-Complex

Complex-valued tensor support for PyTorch. (Work in progress)

Warning: This package is at very early stage, do not use it.

Usage

Warning: this package requires a fresh build of PyTorch revision 6cb593b88cb0c411690b4957850058329526d87b. Other revisions may work, but you will void the warranty.

To use this commit, git clone pytorch and checkout to this commit, then build pytorch from soruce. After you build PyTorch successfully, you will be able to build this plugin just like a normal Python package:

python setup.py install
python setup.py build
python setup.py test

from torch_complex import torch

or

import torch_complex.torch as torch

then the complex tensor support will be in torch module. Use it just like the other tensor types.

Contribution

Please read Pytorch/#755 first.