sbarratt/torch_jacobian.py

Created May 9, 2019 19:40

Star (76) You must be signed in to star a gist
Fork (8) You must be signed in to fork a gist

Select an option

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/sbarratt/37356c46ad1350d4c30aefbd488a4faa.js"></script>
Save sbarratt/37356c46ad1350d4c30aefbd488a4faa to your computer and use it in GitHub Desktop.

Download ZIP

Revisions

sbarratt created this gist May 9, 2019.

torch_jacobian.py

-Original file line number
+Diff line change
@@ -0,0 +1,8 @@
+    def get_jacobian(net, x, noutputs):
+        x = x.squeeze()
+        n = x.size()[0]
+        x = x.repeat(noutputs, 1)
+        x.requires_grad_(True)
+        y = net(x)
+        y.backward(torch.eye(noutputs))
+        return x.grad.data