Mila (Canada) interview question

How do you define a custom layer with a hand crafted gradient function