Skip to content

Add complex autograd support for torch.cumprod #53183

@anjali411

Description

@anjali411
In [9]: x=torch.randn(2, 3, dtype=torch.cdouble, requires_grad=True)

In [10]: torch.cumprod(x, dim=0)
---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
<ipython-input-11-9aad052c3955> in <module>
----> 1 torch.cumprod(x, dim=0)

RuntimeError: cumprod does not support automatic differentiation for outputs with complex dtype.

cc @ezyang @anjali411 @dylanbespalko @mruberry

Metadata

Metadata

Assignees

Labels

complex_autogradmodule: complexRelated to complex number support in PyTorchtriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions