https://camera-optical-zoom.blogspot.com/2023/12/using-op-inputs-when-defining-custom.html
Using Op Inputs When Defining Custom Gradients In Tensorflow