Switch to torch.linalg.norm (#3984)

Commit

3 years ago

Switch to torch.linalg.norm (#3984) * Switch to torch.linalg.norm * Fix formatting