-
Notifications
You must be signed in to change notification settings - Fork 26.1k
Closed
Labels
complex_autogradmodule: complexRelated to complex number support in PyTorchRelated to complex number support in PyTorchmodule: testingIssues related to the torch.testing module (not tests)Issues related to the torch.testing module (not tests)triagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
Description
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.
Metadata
Metadata
Assignees
Labels
complex_autogradmodule: complexRelated to complex number support in PyTorchRelated to complex number support in PyTorchmodule: testingIssues related to the torch.testing module (not tests)Issues related to the torch.testing module (not tests)triagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module