Feature/mps sampling (#28)

* New features implemented. Not debugged yet. * Refactored get_amplitude, added get_statevector and corresponding tests. * Tested and debugged expected_value feature * Added tests for the measure and sample features. Fixed a bug in measure. * Updated the docs. * Changes suggested by Iakov. * Updated the Jupyter notebook, showing how to use the sampling feature.
CQCL · Aug 31, 2023 · 2680f32 · 2680f32
1 parent a3ee4f0
commit 2680f32
Show file tree

Hide file tree

Showing 6 changed files with 655 additions and 99 deletions.
diff --git a/docs/modules/mps.rst b/docs/modules/mps.rst
@@ -12,8 +12,6 @@ Simulation
 
 .. autofunction:: pytket.extensions.cutensornet.mps.simulate
 
-.. autofunction:: pytket.extensions.cutensornet.mps.get_amplitude
-
 
 Classes
 ~~~~~~~
@@ -24,6 +22,13 @@ Classes
     .. automethod:: apply_gate
     .. automethod:: vdot
     .. automethod:: canonicalise
+    .. automethod:: sample
+    .. automethod:: measure
+    .. automethod:: postselect
+    .. automethod:: expectation_value
+    .. automethod:: get_statevector
+    .. automethod:: get_amplitude
+    .. automethod:: get_qubits
     .. automethod:: get_virtual_dimensions
     .. automethod:: get_physical_dimension
     .. automethod:: get_device_id

diff --git a/examples/mps_tutorial.ipynb b/examples/mps_tutorial.ipynb
diff --git a/pytket/extensions/cutensornet/mps/__init__.py b/pytket/extensions/cutensornet/mps/__init__.py
@@ -33,4 +33,4 @@
     MPSxMPO,
 )
 
-from .simulation import ContractionAlg, simulate, get_amplitude, prepare_circuit
+from .simulation import ContractionAlg, simulate, prepare_circuit