gdZddlmZddlmZddlZddlmZddl m Z ddl m Z m Z mZmZdd lmZdd lmZdd lmZGd d e ZdS)z Statevector Estimator V2 class ) annotations)IterableN) SparsePauliOp)BaseEstimatorV2)DataBinEstimatorPubLikePrimitiveResult PubResult) EstimatorPub) PrimitiveJob)_statevector_from_circuitcpeZdZdZddddd Zedd Zedd Zdd ddZddZ d dZ dS)!StatevectorEstimatora" Simple implementation of :class:`BaseEstimatorV2` with full state vector simulation. This class is implemented via :class:`~.Statevector` which turns provided circuits into pure state vectors. These states are subsequently acted on by :class:~.SparsePauliOp`, which implies that, at present, this implementation is only compatible with Pauli-based observables. Each tuple of ``(circuit, observables, parameter values, precision)``, called an estimator primitive unified bloc (PUB), produces its own array-based result. The :meth:`~.EstimatorV2.run` method can be given a sequence of pubs to run in one call. .. note:: The result of this class is exact if the circuit contains only unitary operations. On the other hand, the result could be stochastic if the circuit contains a non-unitary operation such as a reset for a some subsystems. The stochastic result can be made reproducible by setting ``seed``, e.g., ``StatevectorEstimator(seed=123)``. .. plot:: :alt: Output from the previous code. :include-source: from qiskit.circuit import Parameter, QuantumCircuit from qiskit.primitives import StatevectorEstimator from qiskit.quantum_info import Pauli, SparsePauliOp import matplotlib.pyplot as plt import numpy as np # Define a circuit with two parameters. circuit = QuantumCircuit(2) circuit.h(0) circuit.cx(0, 1) circuit.ry(Parameter("a"), 0) circuit.rz(Parameter("b"), 0) circuit.cx(0, 1) circuit.h(0) # Define a sweep over parameter values, where the second axis is over # the two parameters in the circuit. params = np.vstack([ np.linspace(-np.pi, np.pi, 100), np.linspace(-4 * np.pi, 4 * np.pi, 100) ]).T # Define three observables. Many formats are supported here including # classes such as qiskit.quantum_info.SparsePauliOp. The inner length-1 # lists cause this array of observables to have shape (3, 1), rather # than shape (3,) if they were omitted. observables = [ [SparsePauliOp(["XX", "IY"], [0.5, 0.5])], [Pauli("XX")], [Pauli("IY")] ] # Instantiate a new statevector simulation based estimator object. estimator = StatevectorEstimator() # Estimate the expectation value for all 300 combinations of # observables and parameter values, where the pub result will have # shape (3, 100). This shape is due to our array of parameter # bindings having shape (100,), combined with our array of observables # having shape (3, 1) pub = (circuit, observables, params) job = estimator.run([pub]) # Extract the result for the 0th pub (this example only has one pub). result = job.result()[0] # Error-bar information is also available, but the error is 0 # for this StatevectorEstimator. result.data.stds # Pull out the array-based expectation value estimate data from the # result and plot a trace for each observable. for idx, pauli in enumerate(observables): plt.plot(result.data.evs[idx], label=pauli) plt.legend() gN)default_precisionseedrfloatr np.random.Generator | int | Nonec"||_||_dS)a  Args: default_precision: The default precision for the estimator if not specified during run. seed: The seed or Generator object for random number generation. If None, a random seeded default RNG will be used. N)_default_precision_seed)selfrrs ]/granary/edev/quantum/lib/python3.11/site-packages/qiskit/primitives/statevector_estimator.py__init__zStatevectorEstimator.__init__qs#4 returnc|jS)zReturn the default precision)rrs rrz&StatevectorEstimator.default_precision}s &&rc|jS)zAReturn the seed or Generator object for random number generation.)rrs rrzStatevectorEstimator.seeds zr) precisionpubsIterable[EstimatorPubLike]r float | None(PrimitiveJob[PrimitiveResult[PubResult]]c|jfd|D}t|j|}||S)Nc:g|]}tj|S)r coerce).0pubr s r z,StatevectorEstimator.run..s&LLL +C;;LLLr)rr _run_submit)rr!r coerced_pubsjobs ` rrunzStatevectorEstimator.runsO  /ILLLLtLLL 49l33  rlist[EstimatorPub]PrimitiveResult[PubResult]cBtfd|DddiS)Nc:g|]}|Sr')_run_pub)r)r*rs rr+z-StatevectorEstimator._run..s%CCCs c 2 2CCCrversionmetadata)r )rr!s` rr,zStatevectorEstimator._runs0CCCCdCCCyZ[n]]]]rr*r r cdtj|j}|j}|j}|j}|j}||}tj ||\}} tj |tj } tj |tj } tj |j D]} || } | | }t| |}t|\}}t#||}tj||}|dkr9tj|st+d|||}|| | <t/| | | j }t1|||jjdS)N)dtyperz:Given operator is not Hermitian and noise cannot be added.)evsstdsshape)target_precisioncircuit_metadatar8)nprandom default_rngrcircuit observablesparameter_valuesr bind_allbroadcast_arrays zeros_likefloat64ndindexr>rzipitemsr real_if_closeexpectation_valueisreal ValueErrornormalrr r9)rr*rngrDrErFr bound_circuits bc_circuitsbc_obsr<r=index bound_circuit observable final_statepauliscoeffsobsrOdatas rr5zStatevectorEstimator._run_pubsi##DJ//+o /M )227;; 1.+NN VmKrz:::}[ ;;;Z!23 + +E'.MJ3M3GGK *"2"2"4"45NFF//C " 01N1Ns1S1S T T A~~y!233c$%abbb$'JJ/@)$L$L!*CJJ3T;;;  s{Ocdd    r)rrrr)rr)rr)r!r"r r#rr$)r!r1rr2)r*r rr ) __name__ __module__ __qualname____doc__rpropertyrrr0r,r5r'rrrrsOOd-0Z^      '''X'X NR      ^^^^      rr)rb __future__rcollections.abcrnumpyrAqiskit.quantum_inforbaser containersrr r r containers.estimator_pubr primitive_jobr utilsrrr'rrrms#"""""$$$$$$------!!!!!!MMMMMMMMMMMM222222'''''',,,,,,P P P P P ?P P P P P r