Fix vjpvmap for linalg.svdvals (#72811)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/72811
See https://github.com/pytorch/pytorch/pull/72309/files#r803041089 for context
Test Plan: Imported from OSS
Reviewed By: samdow
Differential Revision: D34222388
Pulled By: soulitzer
fbshipit-source-id: 4d1469eb4eed0e36ee87ca99dcc5098835aa9cef
(cherry picked from commit d1f8f111628ac365a2f48a67a957b640a50c4485)
