item for diag batched

pasqal-io · Sep 20, 2024 · 27285d0 · 27285d0
1 parent 582e2b4
commit 27285d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyqtorch/utils.py b/pyqtorch/utils.py
@@ -245,7 +245,7 @@ def is_diag_batched(H: Operator, atol: Tensor = ATOL, batch_dim: int = -1) -> bo
             [is_diag(H[..., i], atol) for i in range(H.shape[batch_dim])],
             device=H.device,
         )
-        return bool(torch.prod(diag_check))
+        return bool(torch.prod(diag_check).item())
     else:
         return is_diag(H, atol)