Enable dtype arg for torch.linalg.norm with order 'fro' and 'nuc' (#46637)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/46255
Pull Request resolved: https://github.com/pytorch/pytorch/pull/46637
Reviewed By: gchanan
Differential Revision: D24459097
Pulled By: mruberry
fbshipit-source-id: 7f207a23de902c27f8313ee80f452687a97e8f6f