I’m new to Zygote.jl and I’d like to compute the squared sum of a gradient, i.e., g = gradient(f, x), q = gradient(g’*g, x) Sucessfully did this using Forward mode: function grad_g_sqr(f, d, x) function gs(x) a = ForwardDiff.gradient(f, x) return a' * a / 2 end gg = F…

Compute gradient of gradient norm using zygote

ToucheSir April 20, 2022, 11:48pm 10

1 Like

Topic		Replies	Views
Newbie: Gradient of a gradient performance in Zygote General Usage zygote	2	506	March 21, 2021
Zygote hessian New to Julia zygote	3	1696	August 10, 2020
Need some help in understanding zygote gradient Machine Learning	2	403	September 7, 2022
Zygote differentiation issues Machine Learning	7	1053	January 28, 2021
Zygote cannot take gradient of a sum when the sum has negative number raised to integer powers? New to Julia	3	707	December 23, 2019