Switch to torch.linalg.norm (#3984)

Commit

2 years ago

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