https://mingredbestcheapprice.blogspot.com/2023/07/function-gradients-with-gather.html