add PEPS.zeros

Author: jcmgray

quimb/tensor/tensor_2d.py

@@ -5008,6 +5008,40 @@ def ones(cls, Lx, Ly, bond_dim, phys_dim=2, like="numpy", **peps_opts):
             **peps_opts,
         )
 
+    @classmethod
+    def zeros(cls, Lx, Ly, bond_dim, phys_dim=2, like="numpy", **peps_opts):
+        """Create a 2D PEPS whose tensors are filled with zeros.
+
+        Parameters
+        ----------
+        Lx : int
+            The number of rows.
+        Ly : int
+            The number of columns.
+        bond_dim : int
+            The bond dimension.
+        physical : int, optional
+            The physical index dimension.
+        peps_opts
+            Supplied to :class:`~quimb.tensor.tensor_2d.PEPS`.
+
+        Returns
+        -------
+        psi : PEPS
+
+        See Also
+        --------
+        PEPS.from_fill_fn
+        """
+        return cls.from_fill_fn(
+            lambda shape: ar.do("zeros", shape, like=like),
+            Lx,
+            Ly,
+            bond_dim,
+            phys_dim,
+            **peps_opts,
+        )
+
     @classmethod
     def rand(
         cls,

tests/test_tensor/test_tensor_2d.py

@@ -53,6 +53,12 @@ def test_cyclic_edge_cases(self):
         assert peps.is_cyclic_y()
         assert peps.num_indices == peps.num_tensors * 3
 
+    def test_zeros(self):
+        peps = qtn.PEPS.zeros(3, 3, cyclic=True, bond_dim=1)
+        assert peps.num_tensors == 9
+        assert peps.num_indices == 27
+        assert_allclose(peps.to_dense(), np.zeros([512, 1]))
+
     def test_flatten(self):
         psi = qtn.PEPS.rand(3, 5, 3, seed=42)
         norm = psi.H & psi