commit c3755eeeee4bbddec0d1399c889e524770182e98
author Gregory Chanan <gchanan@fb.com> Fri Apr 19 10:57:46 2019 -0700
committer Facebook Github Bot <facebook-github-bot@users.noreply.github.com> Fri Apr 19 11:03:14 2019 -0700
tree 01f1434b3688feb55f2c04bd8ce7a9cb37f2dd50
parent 622cf1fec94d284b8974d8c8319aa9199974734d

Make one_hot non-differentiable. (#19430)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19430
ghimport-source-id: 6787473873fdc21400138a4322e17fee8db62607

Differential Revision: D15003382

Pulled By: gchanan

fbshipit-source-id: e9244c7a5f0ad7cd2f79635944a8b37f910231c9

tools/autograd/derivatives.yaml

diff --git a/tools/autograd/derivatives.yaml b/tools/autograd/derivatives.yaml
index 50e3226..d1d8ef7 100644
--- a/tools/autograd/derivatives.yaml
+++ b/tools/autograd/derivatives.yaml
@@ -817,6 +817,9 @@
 - name: t(Tensor self)
   self: grad.t()
 
+- name: one_hot(Tensor self, int64_t num_classes)
+  self: non_differentiable
+
 - name: flip(Tensor self, IntArrayRef dims)
   self: grad.flip(dims)