Issue description

As an enhancement, add an option to QR decomposition to return full matrices, similar to numpy's mode="complete" and tensorflow's full_matrices=True