diff options
| -rw-r--r-- | python/multired.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/python/multired.py b/python/multired.py index 06fbc79..69f99af 100644 --- a/python/multired.py +++ b/python/multired.py @@ -160,7 +160,7 @@ class layer: def compute_VN_entropy_approx(self, poly): p = poly.degree h = - poly[p] * self.N - M = csr_matrix(np.eye(self.N)) + M = csr_matrix(eye(self.N)) for i in range(p-1, -1, -1): M = M * self.resc_laplacian h += - poly[i] * sum(M.diagonal()) |