Feywell:

NotImplementedError


You also need to define backward() function similar to forward (refer here: https://pytorch.org/tutorials/beginner/examples_autograd/two_layer_net_custom_function.html 82)