https://fairytalefashionheadband.blogspot.com/2024/05/gradients-error-using-tensorarray.html