Skip to content

Add complex autograd support for torch.Tensor.index #53605

@anjali411

Description

@anjali411
>>> x = torch.ones(size=(10, 10), dtype=torch.cdouble, requires_grad=True)
>>> y=x.clone()
>>> mask = torch.zeros(size=(10, ), dtype=torch.bool)
>>> y.index_put_((mask, ), y[mask], accumulate=True)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
RuntimeError: index does not support automatic differentiation for outputs with complex dtype.

cc @ezyang @anjali411 @dylanbespalko @mruberry @aocsa

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