Convert to double befoe computing torch norm

Author: kachayev

ot/backend.py

@@ -1882,7 +1882,7 @@ def power(self, a, exponents):
         return torch.pow(a, exponents)
 
     def norm(self, a, axis=None):
-        return torch.linalg.norm(a, dim=axis)
+        return torch.linalg.norm(a.double(), dim=axis)
 
     def any(self, a):
         return torch.any(a)