let elastic_grad a x =   let g1 = l1_grad x in   let g2 = l2_grad x in   MX.(a $* g1 +@ (a $* g2))