answered 3 years ago
You can get much shorter expressions for the components of u_gradient by running
u_gradient
u_gradient.apply_map(factor)
which factorizes the components.