https://purchaseironmanelliptical.blogspot.com/2024/08/tensorflow-differentiate-only-element.html