Learning of diagonal transformation.
- Gradient step on L moved to auxiliar function.