gdZddlmZddlmZddlZddlmZddl m Z ddl m Z dd l mZdd lmZ dddZ dddZdZdZdZdS) z Random state generation. ) annotations)LiteralN) default_rng) QiskitError)random_unitary) Statevector) DensityMatrixdims int | tupleseed int | np.random.Generator | Nonereturnr c|tj}n1t|tjjr|}nt|}tj|}||t}|d||zz }|tj |z}t||S)a~Generator a random Statevector. The statevector is sampled from the uniform distribution. This is the measure induced by the Haar measure on unitary matrices. Args: dims (int or tuple): the dimensions of the state. seed (int or np.random.Generator): Optional. Set a fixed seed or generator for RNG. Returns: Statevector: the random statevector. Reference: K. Zyczkowski and H. Sommers (2001), "Induced measures in the space of mixed quantum states", `J. Phys. A: Math. Gen. 34 7111 `__. N?r ) nprandomr isinstance Generatorprodstandard_normalastypecomplexlinalgnormr )r r rngdimvecs W/granary/edev/quantum/lib/python3.11/site-packages/qiskit/quantum_info/states/random.pyrandom_statevectorr!s( |i##%% D")- . . $ '$--C  c " " ) )' 2 2C2##C(( ((C29>>#  C s & & &&Hilbert-Schmidtrank int | Nonemethod#Literal['Hilbert-Schmidt', 'Bures']r ctj|}||}|dkrt|||}n*|dkrt|||}nt d|t ||S)aGenerator a random DensityMatrix. Args: dims (int or tuple): the dimensions of the DensityMatrix. rank (int or None): Optional, the rank of the density matrix. The default value is full-rank. method (string): Optional. The method to use. 'Hilbert-Schmidt': (Default) sample from the Hilbert-Schmidt metric. 'Bures': sample from the Bures metric. seed (int or np.random.Generator): Optional. Set a fixed seed or generator for RNG. Returns: DensityMatrix: the random density matrix. Raises: QiskitError: if the method is not valid. Nr#BureszError: unrecognized method r)rr_random_density_hs_random_density_buresrr )r r$r&r rrhos r random_density_matrixr-?s2 '$--C | """ dD11 7  #Ct44@@@AAA 4 ( ( ((r"c|tj}n1t|tjjr|}nt|}|||f|||fdzz}|S)a}Return a normally distributed complex random matrix. Args: nrow (int): number of rows in output matrix. ncol (int): number of columns in output matrix. seed(int or np.random.Generator): default rng. Returns: ndarray: A complex rectangular matrix where each real and imaginary entry is sampled from the normal distribution. N)sizer)rrrrrnormal)nrowncolr rginibres r _ginibre_matrixr4es |i##%% D")- . . $jjtTlj++cjjtTlj.K.Kb.PPG Nr"ct|||}||j}|t j|z S)ak Generate a random density matrix from the Hilbert-Schmidt metric. Args: dim (int): the dimensions of the density matrix. rank (int or None): the rank of the density matrix. The default value is full-rank. seed (int or np.random.Generator): default rng. Returns: ndarray: rho (N,N) a density matrix. )r4dotconjTrtrace)rr$r mats r r*r*|sC #tT * *C ''#((**,  C # r"c&tj|t||jz}|t |||}||j}|tj|z S)aWGenerate a random density matrix from the Bures metric. Args: dim (int): the length of the density matrix. rank (int or None): the rank of the density matrix. The default value is full-rank. seed (int or np.random.Generator): default rng. Returns: ndarray: rho (N,N) a density matrix. )r ) reyerdatar6r4r7r8r9)rr$r densityr:s r r+r+sqfSkkN3T:::??G ++oc466 7 7C ''#((**,  C # r")N)r r r rrr )Nr#N) r r r$r%r&r'r rrr )__doc__ __future__rtypingrnumpyr numpy.randomrqiskit.exceptionsr$qiskit.quantum_info.operators.randomr statevectorr densitymatrixr r!r-r4r*r+r"r rIs#"""""$$$$$$))))))??????$$$$$$((((((AE'''''H2C-1 #)#)#)#)#)L.$r"