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