Function cmat_to_numpy

python/runtime/cudaq/operators/py_helpers.cpp:49–61

}

nanobind::object cmat_to_numpy(complex_matrix &cmat) {
  auto rows = cmat.rows();
  auto cols = cmat.cols();
  auto *data = cmat.get_data(complex_matrix::order::row_major);

  // Use .cast() to force immediate creation of the numpy array.
  // Since no owner is specified, rv_policy::automatic will copy the data,
  // making this safe even when cmat is a temporary (e.g. in get_unitary).
  return nanobind::ndarray<nanobind::numpy, std::complex<double>,
                           nanobind::shape<-1, -1>>(data, {rows, cols},
                                                    nanobind::handle())
      .cast();
};

} // namespace cudaq::detail