How to call only backward path of pytorch function?

It looks good. Thanks!