Update A with Procrustes-like solution

Using QR decomposition doesn't seem theoretically guaranteed to decrease the loss