pytorch
db0e121b - [composite compliance] put, take (#81094)

Commit
2 years ago
[composite compliance] put, take (#81094) Reference: #69991 This PR makes `put` CompositeExplicit as it is implemented in terms of `put_` (for which we can't handle Composite Compliance at the implementation level). Ref (put implementation) https://github.com/pytorch/pytorch/blob/478081c69851dbb6b7c570cbcc951199a80df0ff/aten/src/ATen/native/TensorAdvancedIndexing.cpp#L619-L621 Also, we update the `take` gradient formula to handle Tensor Subclass . Pull Request resolved: https://github.com/pytorch/pytorch/pull/81094 Approved by: https://github.com/zou3519
Author
Committer
Parents
Loading