def image_gradients()

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"))