Build a custom layer of calculating gradient in Keras

Question

I want to create a custom layer for calculating the gradient in Keras. I wrote the custom layer but it failed. The concept is followed by grad-cam, I plan to use the gradient ma