Using the gradients in a loss function

[nima_rafiee]

take look at Assign manual assigned "grad_output"