gnEdZddlmZddlZddlmZddlZddlm Z ddl m Z ddl m Z ddlmZdd lmZGd d ZdS) z Abstract QuantumState class. ) annotationsN)abstractmethod) BaseOperator)QuantumChannel)OpShape)Operator)Countsc8eZdZdZdAdBdZdZdZedZed Z ed Z dAd Z d Z dAd Z edCdZedZedZedZedZedDdZedDdZdZdZedAdEdZedAdFd!ZedCdGd%ZdCdHd'ZdAdId*ZdAdJd,ZdAdKd.Ze dLdMd4Z edNd5Z!edNd6Z"e dAdOd8Z#d9Z$d:Z%d;Z&d<Z'd=Z(d>Z)d?Z*d@Z+dS)P QuantumStatez!Abstract quantum state base classNop_shapeOpShape | Nonec"||_d|_dS)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)selfr s ^/granary/edev/quantum/lib/python3.11/site-packages/qiskit/quantum_info/states/quantum_state.py__init__zQuantumState.__init__!s""ct||jo)||kSN) isinstance __class__dimsrothers r__eq__zQuantumState.__eq__3s.%00PTYY[[EJJLL5PPrc&|jjdS)zReturn total state dimension.r)rshapers rdimzQuantumState.dim6s~#A&&rc|jjS)zAReturn the number of qubits if a N-qubit state or None otherwise.)r num_qubitsr s rr#zQuantumState.num_qubits;s~((rcZ|jtjS|jSr)rnprandom default_rngr s r_rngzQuantumState._rng@s(   &9((** *""rc6|j|S)z9Return tuple of input dimension for specified subsystems.)rdims_l)rqargss rrzQuantumState.dimsFs~$$U+++rc*tj|S)z Make a copy of current operator.)copydeepcopyr s rr-zQuantumState.copyJs}T"""rc| d|_dSt|tjjr ||_dStj||_dS)z'Set the seed for the quantum state RNG.N)rrr%r& Generatorr')rvalues rseedzQuantumState.seedNsX ="&D    ry2 3 3 ?"'D   "$)"7"7">">D   rcdS)z%Return True if a valid quantum state.N)ratolrtols ris_validzQuantumState.is_validW  rcdS)z&Convert state to matrix operator classNr4r s r to_operatorzQuantumState.to_operator\r8rcdS)z%Return the conjugate of the operator.Nr4r s r conjugatezQuantumState.conjugatear8rcdS)z:Return the trace of the quantum state as a density matrix.Nr4r s rtracezQuantumState.tracefr8rcdS)z'Return the purity of the quantum state.Nr4r s rpurityzQuantumState.puritykr8rrreturncdS)u$Return 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. Nr4rs rtensorzQuantumState.tensorp  rcdS)u!Return 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. Nr4rs rexpandzQuantumState.expandrDrc@tt|d)aReturn 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_addzQuantumState._adds!"T$ZZ"K"K"KLLLrc@tt|d)aQReturn 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 multiplicationrHrs r _multiplyzQuantumState._multiplys!"T$ZZ"X"X"XYYYrOperator | QuantumChannelr+ list | NonecdS)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. Nr4)rrr+s revolvezQuantumState.evolve roperr None | listcomplexcdS)a Compute 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. Nr4)rrSr+s rexpectation_valuezQuantumState.expectation_values  rdecimals None | int np.ndarraycdS)a)Return 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. Nr4rr+rXs r probabilitieszQuantumState.probabilitiesrRrdictc||||||dS)aaReturn 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. )r+rXT string_labels)_vector_to_dictr]rr\s rprobabilities_dictzQuantumState.probabilities_dictsI(##   UX  > > IIe  $   rshotsintc||}|tjt |||d}|j|||S)afSample 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. Tr`psize)r]_index_to_ket_arrayr%arangelenrr(choice)rrdr+probslabelss r sample_memoryzQuantumState.sample_memorysu.""5)))) Ic%jj ! !499U#3#34*  y%e<<>**6499U3C3CSW*XXYZ[xE '222275=111V kk(274==TtTTT\akbb|rFrwrraboolcdg}|ddD] }||d|z!tjfdt||D}|rt |}tj|tj}|d}|ddD]H} |dkr tjd|}tj| |}I|j S|j S) 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. r|Nc&g|] \}}|z|zSr4r4).0r!shiftrws r z4QuantumState._index_to_ket_array..js&SSS:3$%-3.SSSrr}r ,) appendr%arrayrumaxasarraystr_charaddT) rwrrashiftsr!ketsmax_dim char_ketsstr_ketsrows ` rrjz QuantumState._index_to_ket_arrayWs 9 , ,C MM&*s* + + + +xSSSST6ARARSSSTT  $iiG 4rw777I |H } 6 6R<<!w{{399H7;;sH55: v rc&||n||}|\}t|||}|r#t t |||Sdt |||DS)aConvert 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. NrXr`c4i|]\}}t||Sr4rz)rketvals r z0QuantumState._vector_to_dict..s$FFFHCc CFFFr)roundnonzeror rjr^ru)vecrrXravalsrwrs rrbzQuantumState._vector_to_dictxs&&ssCIIxI,H,H,,..//d-/XX  .D#d),,-- -FFD$t*0E0EFFFFrch||n||}|\}}t|||}t|||}|r#dt |||||fDSdt |||||fDS)aConvert 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. Nrr`c&i|]\}}}|d||S)|r4rrbrars rrz0QuantumState._matrix_to_dict..s7(5S#3rcTi|]%\}}}t|t|f|&Sr4rrs rrz0QuantumState._matrix_to_dict..sA   S#3ZZs $c   r)rrr rjru) matrrXrarinds_rowinds_colbrasrs r_matrix_to_dictzQuantumState._matrix_to_dicts&&ssCIIxI,H,H LLNN  //$m/\\//$m/\\  9.s!<<z8QuantumState._subsystem_probabilities..s-GGq1J3F3F3F3F3F3FGGr)axis)axes) r%reshapelistreversedrrzrangesumargsort transposeri)rnrr+ probs_tenssum_axis new_probsrrs @@r_subsystem_probabilitiesz%QuantumState._subsystem_probabilitiess" =LZtHTNN';';<< <<<>> JzJO+=>> rc,||Sr)rQrs r__and__zQuantumState.__and__{{5!!!rc,||Sr)rCrs r__xor__zQuantumState.__xor__rrc,||SrrMrs r__mul__zQuantumState.__mul__s~~e$$$rc2|d|z S)Nr|rrs r __truediv__zQuantumState.__truediv__s~~a%i(((rc,||Sr)rrs r__rmul__zQuantumState.__rmul__s||E"""rc,||SrrKrs r__add__zQuantumState.__add__syyrc.|| Srrrs r__sub__zQuantumState.__sub__syy%   rc,|dS)Nrrr s r__neg__zQuantumState.__neg__s~~b!!!rr)r r )NN)rr rAr )rrNr+rOrAr )rSrr+rTrArU)r+rTrXrYrArZ)r+rTrXrYrAr^)rdrer+rTrArZ)rdrer+rTrAr )r+rOrArz)F)rwrZrrzrarrArZ)NF)rnrZrrzr+rTrArZ),__name__ __module__ __qualname____doc__r__array_priority__rpropertyr!r#r(rr-r2rr7r:r<r>r@rCrFrKrMrQrWr]rcrpryr staticmethodrjrbrrrrrrrrrrr4rrr r s++ # # # # # QQQ''X'))X)##X# ,,,,###????   ^   ^   ^   ^   ^     ^      ^   M M M Z Z Z    ^ "     ^      ^ "     4=====>))))):#####J=B\@GGG\G<% % % \% N=A\B""""""%%%)))###   !!!"""""rr )r __future__rr-abcrnumpyr%+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 r4rrrs#""""" DDDDDDPPPPPP::::::;;;;;;''''''Y"Y"Y"Y"Y"Y"Y"Y"Y"Y"r