scale MPS

block-hczhai · Sep 25, 2024 · 74d18d2 · 74d18d2
1 parent ce6d196
commit 74d18d2
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/src/dmrg/mps.hpp b/src/dmrg/mps.hpp
@@ -2385,6 +2385,12 @@ template <typename S, typename FL> struct MPS {
         for (int i = 0; i < n_sites; i++)
             random_canonicalize_tensor(i);
     }
+    virtual void iscale(FL d) {
+        load_tensor(center);
+        tensors[center]->iscale(d);
+        save_tensor(center);
+        unload_tensor(center);
+    }
     virtual void set_inact_ext_identity(int n_inactive, int n_external) {
         for (int i = 0; i < n_inactive; i++)
             if (tensors[i] != nullptr && i < center) {

diff --git a/src/pybind/pybind_core.hpp b/src/pybind/pybind_core.hpp
@@ -874,6 +874,7 @@ template <typename S, typename FL> void bind_fl_sparse(py::module &m) {
              py::arg("length"))
         .def("trace", &SparseMatrix<S, FL>::trace)
         .def("norm", &SparseMatrix<S, FL>::norm)
+        .def("iscale", &SparseMatrix<S, FL>::iscale)
         .def("conjugate", &SparseMatrix<S, FL>::conjugate)
         .def("__getitem__",
              [](SparseMatrix<S, FL> *self, int idx) { return (*self)[idx]; })

diff --git a/src/pybind/pybind_dmrg.hpp b/src/pybind/pybind_dmrg.hpp
@@ -408,6 +408,7 @@ template <typename S, typename FL> void bind_fl_mps(py::module &m) {
         .def("canonicalize", &MPS<S, FL>::canonicalize)
         .def("dynamic_canonicalize", &MPS<S, FL>::dynamic_canonicalize)
         .def("random_canonicalize", &MPS<S, FL>::random_canonicalize)
+        .def("iscale", &MPS<S, FL>::iscale)
         .def("set_inact_ext_identity", &MPS<S, FL>::set_inact_ext_identity)
         .def("from_singlet_embedding_wfn",
              &MPS<S, FL>::from_singlet_embedding_wfn, py::arg("cg"),