backenc.calculate_norm uses numpy.linalg.norm

qiboteam · Oct 12, 2023 · 6dfd2ca · 6dfd2ca
1 parent f45aa26
commit 6dfd2ca
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/qibo/backends/numpy.py b/src/qibo/backends/numpy.py
@@ -673,13 +673,14 @@ def entanglement_entropy(self, rho):
         entropy = self.np.sum(masked_eigvals * spectrum) / self.np.log(2.0)
         return entropy, spectrum
 
-    def calculate_norm(self, state):
+    def calculate_norm(self, state, order=None):
+        """"""
         state = self.cast(state)
-        return self.np.sqrt(self.np.sum(self.np.abs(state) ** 2))
+        return self.np.linalg.norm(state, ord=order)
 
-    def calculate_norm_density_matrix(self, state):
+    def calculate_norm_density_matrix(self, state, order="nuc"):
         state = self.cast(state)
-        return self.np.trace(state)
+        return self.np.linalg.norm(state, ord=order)
 
     def calculate_overlap(self, state1, state2):
         state1 = self.cast(state1)