Switch to torch.linalg.norm

[loadams]

See linked issue, but torch.norm is deprecated, so this switches to torch.linalg.norm and renamed changed parameters.

torch.linalg.norm was added in torch 1.7.0.

Function signature changes:

torch.norm(input, p='fro', dim=None, keepdim=False, out=None, dtype=None)
torch.linalg.norm(A, ord=None, dim=None, keepdim=False, *, out=None, dtype=None)