https://camera-optical-zoom.blogspot.com/2023/05/function-gradients-with-gather.html