Can I compute gradient of the model output w.r.t input?

[junhyeokahn]

Hello, First of all, I appreciate your effort to maintain this nice repo. I am wondering if I can get a gradient of the network model output w.r.t input similar to this function in tensorflow.