torch_norm

omnipose.utils.torch_norm(a, dim=0, keepdim=False)

Wrapper for torch.linalg.norm to handle ARM architecture.