Custom Loss Function with Backward Method

autograd
3

Checkout this thread: Difference between apply an call for an autograd function