Skip to content

Add complex autograd support for torch.Tensor.index_put_ #53645

Closed
@anjali411

Description

@anjali411

It also looks like autograd for torch.Tensor.index_put_ is not even being tested for floating point tensors correctly.

>>> mask = torch.zeros(size=(10, ), dtype=torch.bool)
>>> y = torch.ones(size=(10, 10), dtype=torch.cdouble, requires_grad=True)
>>> y.clone().index_put_((mask, ), y[mask])
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

No one assigned

    Labels

    complex_autogradmodule: complexRelated to complex number support in PyTorchmodule: testingIssues related to the torch.testing module (not tests)triagedThis 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