CdddlmZddlmZddlmZddlZddlm Z ddl m Z ddl m Z mZddlmZdd lmZdd lmZdd lmZdd lmZd ZdZdZe d#dZeeZe d$dZdZ e j!ddee ej"dZ#e j!ddee ej$dZ%e j!ddee ej&dZ'eeZ(e(ej'ej&Z&e j!ddee ej)Z*e(ej+ej)Z,e*j-@e*j-.dde*_-e*j-.dd e*_-d!Z+d"Z)e*j-e+_-e,j-e)_-dS)%) annotationspartial)productN)curry)array_creation_dispatch)Arraynormalize_chunks)meta_from_array)tokenize) blockwise)ArrayChunkShapeDep)funcnamec t|tjr|}t|tt fs|f}|dd}|dd}|dd}|||g|Ri|j}tj|}t|||}|p&t|dzt||||||z}|||||dS)Nnamechunksautodtyper-)shaperkwargsrr) isinstancenpndarraytolisttuplelistpoprr rr )funcargsrrrrrs /lib/python3.11/site-packages/dask/array/wrap.py_parse_wrap_argsr#s%$$  eeT] + + ::fd # #D ZZ& ) )F JJw % %E }U,T,,,V,,2 HUOOE fe5 9 9 9F  8D>>C'( eVUD&++D    c d|vr|d|dd}}n|d}t|trtdt ||||}|d}|d}|d}|d}|d }t |fd|i|}t tt|x}} t|||t|| i } t| |||| d d S) = Transform np creation function into blocked version rrNzTDask array input not supported. Please use tuple, list, or a 1D numpy array instead.rrrr) numblocksmeta)rr)) rrr TypeErrorr#rrrangelencore_blockwiserget) r r!rrparsedrrrout_inddep_indgraphs r"wrap_func_shape_as_first_argr30s6f1gtABBxt 7##%  C   dD&% 8 8F 7OE 7OE H F &>D H F 4 / /u / / /DeCJJ//000Gg   6""    E fE 648P8P Q Q QQr$czd}t|}d|j}t|}|d}|d|d}|d}|dt |ggd|DR} t |} t | } fd| D} t t | D]\} } | | | d<fd t| | D}tt| |}t||||  S) r&rrrrrrcFg|]}tt|S)r+r,).0bds r" z"wrap_func_like..cs$===U3r77^^===r$cg|]}Sr6r6)r7_rs r"r9z"wrap_func_like..fs ! ! !Q& ! ! !r$c3DK|]\}}tfdi|fzVdS)rNr)r7ksr!rr s r" z!wrap_func_like..isD T T&1aWT , , ,! , , . 5 T T T T T Tr$)r)) r r.rr#rr enumeratezipdictr astype)r r!rxr)rr/rrkeysshapeskwir>valsdskrs``` @r"wrap_func_likerKTsX QA 1  D JJw ( (E dD&% 8 8F 7OE 7OE H F &>D H F D6 >==f=== > > >D f F &\\F ! ! ! !& ! ! !B$v,,''11g T T T T T TCFOO T T TD s4  C dFU);); < < <>> x = np.broadcast_to(1, (100,100,100)) >>> x.base.nbytes 8 Those array are not only more efficient locally, but dask serialisation is aware of the _real_ size of those array and thus can send them around efficiently and schedule accordingly. Note that those array are read-only and numpy will refuse to assign to them, so should be safe. )rXrNrO)r inners r"broadcast_trickr[s(& #4 ( (ELEM]EN Lr$numpyones)backendrf8rzerosempty)rQfullz*array([0.1, 0.1, 0.1, 0.1, 0.1, 0.1])z%array([0.1, 0.1, 0.1, 0.1, 0.1, 0.1])z >>> np.full_like(y, [0, 0, 255])zB>>> np.full_like(y, [0, 0, 255]) # doctest: +NORMALIZE_WHITESPACEc tj|dkr%tdt|jd|dd-t |dr |j|d<nt||d<t|||d|S)Nr$fill_value must be scalar. Received instead.r)r fill_value) rndim ValueErrortyperOr.hasattrr_full)rrfr!rs r"rbrbs wza W4 3C3C3L W W W   zz'4  ( :w ' ' /(.F7OO":..F7O $e E Ef E EEr$ctj|dkr%tdt|jdt |||d|S)Nrrdre)arf)rrgrhrirO _full_like)rmrfr!rs r" full_likerosn wza W4 3C3C3L W W W           r$)N)r6)/ __future__r functoolsr itertoolsrr\rtlzrdask.array.backendsrdask.array.corer r dask.array.utilsr dask.baser dask.blockwiser r- dask.layersr dask.utilsrr#r3rKrTwrXr[register_inplace ones_liker] zeros_liker` empty_likeraw_likerorkrbrnrNreplacer6r$r"rsv""""""77777733333333,,,,,,666666******