xla
ca7297d3
- Add one_hot to ATen XLA tensor
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
6 years ago
Add one_hot to ATen XLA tensor It's sufficient to just route it to at::native::one_hot. Still goes back to host for range validation of labels, which cannot be avoided.
References
#473 - Add min, max, one_hot
Author
asuhan
Committer
asuhan
Parents
143d89d6
Loading