Implement a Custom `autograd.Function`Deep-ML | Problem