How to get the diagonal hessian

[bicycle]

Hi all,

I’d like to get the sum of diagonal hessian (i.e. laplace) of all parameters in a CNN, is there any efficient way? Until now, I can only compute the full hessian by loop, but it is too time-consuming…