Hi,
Well I don’t know what this green arrow would compute.
But if you have a formula for it, you can create a new autograd Function as described here 11 to define your own backward for a non-differentiable forward.