https://travelebookdownload.blogspot.com/2024/06/eager-execution-gradient-computation.html