A package to easily install tensorflow utils for complex-valued tensors and neural networks
This package was inspired by the work of Elizabeth Cole et al.: Image Reconstruction using an Unrolled DL Architecture including Complex-Valued Convolution and Activation Functions.
Please cite their work appropriately if you use this package.
The code for their work is available here.
You can install tf-complex
using pypi:
pip install tf-complex
You can define a complex convolution in the following way to use in one of your models:
from tf_complex.convolutions import ComplexConv2D
conv = ComplexConv2D(
16,
3,
padding='same',
activation='crelu',
)