Fix `torch.linalg.qr` example (#54342)
Summary:
Since a.size() is (3, 4, 5), so r.size() is (3, 4, 5) , but q.size is (3, 4, 4)
Also, reduce tolerance from 1e-8 to 1e-5 as
Fixes https://github.com/pytorch/pytorch/issues/54320
Pull Request resolved: https://github.com/pytorch/pytorch/pull/54342
Reviewed By: walterddr
Differential Revision: D27193947
Pulled By: malfet
fbshipit-source-id: 362a0fdd90550888a4f0c6deaa49b9f72d379842