in code/custom_hook.py [0:0]
def image_gradients(self, image): """Register input image for backward pass, to get image gradients""" image.register_hook(self.backward_hook("image"))