Add support for integer dim arg in `torch.linalg.norm` (#43907)
Summary:
Since PR https://github.com/pytorch/pytorch/issues/43262 is merged, this works now.
Part of https://github.com/pytorch/pytorch/issues/24802
Pull Request resolved: https://github.com/pytorch/pytorch/pull/43907
Reviewed By: anjali411
Differential Revision: D23471964
Pulled By: mruberry
fbshipit-source-id: ef2f11f78343fc866f752c9691b0c1fa687353ba
