gjdZddlmZddlZddlmZddlmZddl m Z m Z ddl m Z dd lmZmZmZeZdddZ dddZ dddZdS)z% Methods to create random operators. ) annotationsN) default_rng) QiskitError)Operator Stinespring)random_cnotdihedral)random_clifford random_paulirandom_pauli_listdims int | tupleseed int | np.random.Generator | Nonec|t}n1t|tjjr|}nt |}tj|}ddlm}|j ||}t|||S)akReturn a random unitary Operator. The operator is sampled from the unitary Haar measure. Args: dims (int or tuple): the input dimensions of the Operator. seed (int or np.random.Generator): Optional. Set a fixed seed or generator for RNG. Returns: Operator: a unitary operator. Nrstats random_state input_dims output_dims) DEFAULT_RNG isinstancenprandom Generatorrprodscipyr unitary_grouprvsr)r rrdimrmats Z/granary/edev/quantum/lib/python3.11/site-packages/qiskit/quantum_info/operators/random.pyrandom_unitaryr% s |" D")- . .) "4(( '$--C   ! !#L ! A AC CDd ; ; ;;F tracelessboolc|t}n1t|tjjr|}nt |}tj|}ddlm}|rtj ||ft}nGtj |j d||t}||dz zdz}|j d||}|j d||} tj|d \} } |d | zz|| | f<|d | zz || | f<t!||| S) a$Return a random hermitian Operator. The operator is sampled from Gaussian Unitary Ensemble. Args: dims (int or tuple): the input dimension of the Operator. traceless (bool): Optional. If True subtract diagonal entries to return a traceless hermitian operator (Default: False). seed (int or np.random.Generator): Optional. Set a fixed seed or generator for RNG. Returns: Operator: a Hermitian operator. Nrr)dtyper)scalesizerg?y?r)rrrrrrrrrzeroscomplexdiagnormr!astype tril_indicesr) r r'rrngr"rr#num_tril real_tril imag_trilrowscolss r$random_hermitianr;;sQ$ | D")- . . $ '$--C[hSz111gejnn13SnIIPPQXYYZZsQwA%H SxcJJI SxcJJIb))JD$!BN2Ct !BN2Ct  CDd ; ; ;;r&rint | tuple | Nonerrank int | Nonec~||td||}n||}tj|}tj|}| |||zkr||z}|dkrtd|dddlm}|jt||z||}t|ddd|f|| S) aReturn a random CPTP quantum channel. This constructs the Stinespring operator for the quantum channel by sampling a random isometry from the unitary Haar measure. Args: input_dims (int or tuple): the input dimension of the channel. output_dims (int or tuple): the input dimension of the channel. rank (int): Optional. The rank of the quantum channel Choi-matrix. seed (int or np.random.Generator): Optional. Set a fixed seed or generator for RNG. Returns: Stinespring: a quantum channel operator. Raises: QiskitError: if rank or dimensions are invalid. NzHNo dimensions specified: input_dims and output_dims cannot both be None.rzRank z must be greater than 0.rrrr) rrrrrr r!maxr)rrr=rd_ind_outrunitarys r$random_quantum_channelrDis2k1 V      7:  D GK E |tdUl**e| axx@$@@@AAA!%%c$,&=&=D%QQG wqqq%4%x(Z[ Y Y YYr&)N)r rrr)FN)r rr'r(rr)NNNN)rr<rr<r=r>rr)__doc__ __future__rnumpyr numpy.randomrqiskit.exceptionsrqiskit.quantum_info.operatorsrrdihedral.randomr symplectic.randomr r r rr%r;rDr&r$rNs#"""""$$$$$$))))))????????100000OOOOOOOOOOkmm <<<<<8Z^+<+<+<+<+<^&*&*-1 1Z1Z1Z1Z1Z1Z1Zr&