z inESrSSKJr SSKrSSKJr SSKrSSKJ r SSK J r SSK J r SSKJr SS KJr "S S 5rg) z Abstract QuantumState class. ) annotationsN)abstractmethod) BaseOperator)QuantumChannel)OpShape)Operator)Countsc*\rSrSrSrS*S+SjjrSrSr\S5r \S5r \S 5r S*S jr S r S*S jr\S,S j5r\S5r\S5r\S5r\S5r\S-Sj5r\S-Sj5rSrSr\S*S.Sjj5r\S*S/Sjj5r\S,S0Sjj5rS,S1SjjrS*S2SjjrS*S3SjjrS*S4Sjjr\ S5S6Sjj5r!\ S7Sj5r"\ S7Sj5r#\ S*S8S jj5r$S!r%S"r&S#r'S$r(S%r)S&r*S'r+S(r,S)r-g)9 QuantumStatez!Abstract quantum state base classNcXlSUlg)zInitialize a QuantumState object. Args: op_shape (OpShape): Optional, an OpShape object for state dimensions. .. note:: If `op_shape`` is specified it will take precedence over other kwargs. N _op_shape_rng_generator)selfop_shapes b/home/james-whalen/.local/lib/python3.13/site-packages/qiskit/quantum_info/states/quantum_state.py__init__QuantumState.__init__!s""c|[XR5=(a! UR5UR5:H$N) isinstance __class__dimsrothers r__eq__QuantumState.__eq__3s'%0PTYY[EJJL5PPrc4URRS$)zReturn total state dimension.r)rshapers rdimQuantumState.dim6s~~##A&&rc.URR$)zAReturn the number of qubits if a N-qubit state or None otherwise.)r num_qubitsr#s rr'QuantumState.num_qubits;s~~(((rcpURc[RR5$UR$r)rnprandom default_rngr#s r_rngQuantumState._rng@s-    &99((* *"""rc8URRU5$)z9Return tuple of input dimension for specified subsystems.)rdims_l)rqargss rrQuantumState.dimsFs~~$$U++rc.[R"U5$)z Make a copy of current operator.)copydeepcopyr#s rr4QuantumState.copyJs}}T""rcUcSUlg[U[RR5(aXlg[RR U5Ulg)z'Set the seed for the quantum state RNG.N)rrr*r+ Generatorr,)rvalues rseedQuantumState.seedNsC ="&D  ryy22 3 3"' "$))"7"7">D rcg)z%Return True if a valid quantum state.N)ratolrtols ris_validQuantumState.is_validW rcg)z&Convert state to matrix operator classNr=r#s r to_operatorQuantumState.to_operator\rBrcg)z%Return the conjugate of the operator.Nr=r#s r conjugateQuantumState.conjugatearBrcg)z:Return the trace of the quantum state as a density matrix.Nr=r#s rtraceQuantumState.tracefrBrcg)z'Return the purity of the quantum state.Nr=r#s rpurityQuantumState.puritykrBrcg)uReturn the tensor product state self ⊗ other. Args: other (QuantumState): a quantum state object. Returns: QuantumState: the tensor product operator self ⊗ other. Raises: QiskitError: if other is not a quantum state. Nr=rs rtensorQuantumState.tensorp rcg)uReturn the tensor product state other ⊗ self. Args: other (QuantumState): a quantum state object. Returns: QuantumState: the tensor product state other ⊗ self. Raises: QiskitError: if other is not a quantum state. Nr=rs rexpandQuantumState.expandrRrc0[[U5S35e)zReturn the linear combination self + other. Args: other (QuantumState): a state object. Returns: QuantumState: the linear combination self + other. Raises: NotImplementedError: if subclass does not support addition. z does not support additionNotImplementedErrortypers r_addQuantumState._adds"T$ZL0J"KLLrc0[[U5S35e)aReturn the scalar multipled state other * self. Args: other (complex): a complex number. Returns: QuantumState: the scalar multipled state other * self. Raises: NotImplementedError: if subclass does not support scala multiplication. z' does not support scalar multiplicationrWrs r _multiplyQuantumState._multiplys"T$ZL0W"XYYrcg)aEvolve a quantum state by the operator. Args: other (Operator or QuantumChannel): The operator to evolve by. qargs (list): a list of QuantumState subsystem positions to apply the operator on. Returns: QuantumState: the output quantum state. Raises: QiskitError: if the operator dimension does not match the specified QuantumState subsystem dimensions. Nr=)rrr1s revolveQuantumState.evolve rcg)zCompute the expectation value of an operator. Args: oper (BaseOperator): an operator to evaluate expval. qargs (None or list): subsystems to apply the operator on. Returns: complex: the expectation value. Nr=)roperr1s rexpectation_valueQuantumState.expectation_values rcg)aReturn the subsystem measurement probability vector. Measurement probabilities are with respect to measurement in the computation (diagonal) basis. Args: qargs (None or list): subsystems to return probabilities for, if None return for all subsystems (Default: None). decimals (None or int): the number of decimal places to round values. If None no rounding is done (Default: None). Returns: np.array: The Numpy vector array of probabilities. Nr=rr1decimalss r probabilitiesQuantumState.probabilitiesrbrc\URURXS9URU5SS9$)aReturn the subsystem measurement probability dictionary. Measurement probabilities are with respect to measurement in the computation (diagonal) basis. This dictionary representation uses a Ket-like notation where the dictionary keys are qudit strings for the subsystem basis vectors. If any subsystem has a dimension greater than 10 comma delimiters are inserted between integers so that subsystems can be distinguished. Args: qargs (None or list): subsystems to return probabilities for, if None return for all subsystems (Default: None). decimals (None or int): the number of decimal places to round values. If None no rounding is done (Default: None). Returns: dict: The measurement probabilities in dict (ket) form. )r1riT string_labels)_vector_to_dictrjrrhs rprobabilities_dictQuantumState.probabilities_dicts<(##   U  > IIe $  rcURU5nUR[R"[ U55UR U5SS9nUR RXCUS9$)aSample a list of qubit measurement outcomes in the computational basis. Args: shots (int): number of samples to generate. qargs (None or list): subsystems to sample measurements for, if None sample measurement of all subsystems (Default: None). Returns: np.array: list of sampled counts if the order sampled. Additional Information: This function *samples* measurement outcomes using the measure :meth:`probabilities` for the current state and `qargs`. It does not actually implement the measurement so the current state is not modified. The seed for random number generator used for sampling can be set to a fixed value by using the stats :meth:`seed` method. Trmpsize)rj_index_to_ket_arrayr*arangelenrr-choice)rshotsr1probslabelss r sample_memoryQuantumState.sample_memorysd.""5))) IIc%j !499U#34* yye<**699U3CSW*XYZ[xxE '22775=11 kk(2774=TT\akb|rcS/nUSSHnURUSU-5 M [R"[X5VVs/sH upEX-U-PM snn5nU(a[ U5n[R "U[R S9nUSn USSHHn US:a [RRSU 5n [RRX5n MJ U R$UR$s snnf)aConvert an index array into a ket array. Args: inds (np.array): an integer index array. dims (tuple): a list of subsystem dimensions. string_labels (bool): return ket as string if True, otherwise return as index array (Default: False). Returns: np.array: an array of ket strings if string_label=True, otherwise an array of ket lists. rNrr ,) appendr*arrayrmaxasarraystr_charaddT) rrrnshiftsr$shiftketsmax_dim char_ketsstr_ketsrows rrv QuantumState._index_to_ket_arrayWs 9C MM&*s* +xxTARSAR:3$-3.ARST $iG 4rww7I |H }R<!ww{{39H77;;s5%:: vv TsD cUcUOURUS9nUR5un[RXQUS9nU(a[ [ X`U55$[ XdU5VVs0sHupx[ U5U_M snn$s snnf)a6Convert a vector to a ket dictionary. This representation will not show zero values in the output dict. Args: vec (array): a Numpy vector array. dims (tuple): subsystem dimensions. decimals (None or int): number of decimal places to round to. (See Numpy.round), if None no rounding is done (Default: None). string_labels (bool): return ket as string if True, otherwise return as index array (Default: False). Returns: dict: the vector in dictionary `ket` form. rirm)roundnonzeror rvdictrtuple) vecrrirnvalsrrketvals rroQuantumState._vector_to_dictxs&&sCIIxI,H,,.//-/X Dd),- -03Dt*0EF0EHCc C0EFFFs*Bc UcUOURUS9nUR5unn[RXQUS9n[RXaUS9nU(a-[ XXEU45V V V s0sH upoSU 3U _M sn n n $[ XXEU45V V V s0sHupn [ U 5[ U 54U _M sn n n $s sn n n fs sn n n f)a6Convert a matrix to a ket dictionary. This representation will not show zero values in the output dict. Args: mat (array): a Numpy matrix array. dims (tuple): subsystem dimensions. decimals (None or int): number of decimal places to round to. (See Numpy.round), if None no rounding is done (Default: None). string_labels (bool): return ket as string if True, otherwise return as index array (Default: False). Returns: dict: the matrix in dictionary `ket` form. rrm|)rrr rvrr) matrrirnrinds_rowinds_colbrasrrbrars r_matrix_to_dictQuantumState._matrix_to_dicts&&sCIIxI,H LLN  //m/\//m/\ 9[R"X6S9n[R"[R"T55m[R"UTS9n[R"X3R45nU$s snf)aMarginalize a probability vector according to subsystems. Args: probs (np.array): a probability vector Numpy array. dims (tuple): subsystem dimensions. qargs (None or list): a list of subsystems to return marginalized probabilities for. If None return all probabilities (Default: None). Returns: np.array: the marginalized probability vector flattened for the specified qargs. rc36># UHoT;dM Uv M g7frr=).0i qargs_axess r 8QuantumState._subsystem_probabilities..sGKqJ3FKs  )axis)axes) r*reshapelistreversedndimrrangesumargsort transposeru) r{rr1 probs_tensrrsum_axis new_probsrs @r_subsystem_probabilities%QuantumState._subsystem_probabilitiess" =LZZtHTN';< ,4UO JJzOO+=> =s C8c$URU5$r)r`rs r__and__QuantumState.__and__{{5!!rc$URU5$r)rPrs r__xor__QuantumState.__xor__rrc$URU5$rr]rs r__mul__QuantumState.__mul__s~~e$$rc*URSU- 5$)Nrrrs r __truediv__QuantumState.__truediv__s~~a%i((rc$URU5$r)rrs r__rmul__QuantumState.__rmul__s||E""rc$URU5$rrZrs r__add__QuantumState.__add__syyrc&URU*5$rrrs r__sub__QuantumState.__sub__syy%  rc$URS5$)Nrrr#s r__neg__QuantumState.__neg__s~~b!!rrr)rzOpShape | None)NN)rr returnr )rzOperator | QuantumChannelr1 list | Nonerr )rdrr1 None | listrr)r1rri None | intr np.ndarray)r1rrirrr)rzintr1rrr)rzrr1rrr )r1rrr)F)rrrrrnboolrr)NF)r{rrrr1rrr).__name__ __module__ __qualname____firstlineno____doc__r__array_priority__rpropertyr$r'r-rr4r:rr@rDrGrJrMrPrTrZr]r`rerjrpr}rr staticmethodrvrorrrrrrrrrr__static_attributes__r=rrr r s+ # Q''))## ,#?                   M Z  "      " 4=>):#J=B %6: @GG<% % N=A!&/: B""%)# !"rr )r __future__rr4abcrnumpyr*+qiskit.quantum_info.operators.base_operatorr5qiskit.quantum_info.operators.channel.quantum_channelr&qiskit.quantum_info.operators.op_shaper&qiskit.quantum_info.operators.operatorrqiskit.result.countsr r r=rrr s3# DP:;'Y"Y"r