Autograd for complex-valued neural networks

Hi,
I have a doubt for autograd for complex-valued neural networks(Autograd mechanics — PyTorch 1.11.0 documentation).It seems that autograd works when differentiating complex-valued tensors. But does it also work for layers in a neural network?
So lets say I define a F.Linear layer, can we define the weights to be complex-valued and will the updates to this layer follow complex-valued backpropogation? Also, can the same thing be done for nn.Conv2d layer?

Hi Anirudh!

Yes, autograd performs complex backpropagation through (most, if not
all) pytorch layers and with respect to those layers’ parameters.

Consider this Linear example:

>>> import torch
>>> torch.__version__
'1.11.0'
>>> _ = torch.manual_seed (2022)
>>> lin = torch.nn.Linear (2, 2, bias = False, dtype = torch.complex64)
>>> lin.weight
Parameter containing:
tensor([[-0.1474+0.5967j,  0.3660-0.1681j],
        [-0.6700-0.1989j,  0.4149+0.3975j]], requires_grad=True)
>>> t = torch.randn (5, 2, dtype = torch.complex64)
>>> t
tensor([[-0.5672+0.5518j, -0.8078+0.3621j],
        [ 0.0075+0.4234j, -0.4115-0.4382j],
        [-1.5592+0.4851j,  0.6728-0.1796j],
        [ 0.4976+0.4027j, -0.9824-0.7075j],
        [-0.1651-0.3127j, -0.6328-0.4980j]])
>>> lin (t).norm().backward()
>>> lin.weight.grad
tensor([[-0.3748+1.0385j,  0.6620-0.4970j],
        [-1.2673-0.4408j,  0.8804+0.7391j]])

Note, the implementation of complex gradients has built into it, in some
sense, the notion that you are calculating the gradient of a real-valued
loss function.

If you calculate the gradient of a complex “loss” function, you might not
get the results you expect (depending upon your expectations).

Pytorch’s support for complex manipulations is still a work in progress, but
most (if not all) of the basics have been implemented and work correctly.

Best.

K. Frank

Hi K.Frank,
Thanks for replying.I was trying to do the same for convolutional layer.
The following code is a normal convolution operation.

m = torch.nn.Conv2d(16, 33, (3, 3))
input = torch.randn(20, 16, 50, 50)
output = m(input)
print(output)

But when I do it for complex-convolution,

m = torch.nn.Conv2d(16, 33, (3, 3),dtype=torch.complex64)
input = torch.randn(20, 16, 50, 50,dtype=torch.complex64)
output = m(input)
print(output)

It gives me the following error,

RuntimeError: “slow_conv2d_cpu” not implemented for ‘ComplexFloat’.

Hi Anirudh!

You’re right, Conv2d does not yet have complex support in the latest
stable release (1.11.0) – that’s my fault for not double-checking.

(As I said, pytorch’s complex support is still a work in progress.)

However, Conv2d does have complex support in the latest nightly build
(a 1.12 version):

>>> import torch
>>> torch.__version__
'1.12.0.dev20220510'
>>> _ = torch.manual_seed (2022)
>>> conv = torch.nn.Conv2d (1, 1, 3, dtype = torch.complex64)
>>> conv (torch.randn (1, 1, 3, 3, dtype = torch.complex64))
tensor([[[[0.1691-0.2932j]]]], grad_fn=<AddBackward0>)

Best.

K. Frank