Get gradient wrt to input

[jsuit]

Figured it out here: Strange behavior of Variable.cuda() and Variable.grad