U /e@sddlmZddlmZddlZddlmZddlm Z ddl m Z m Z ddl mZddlmZdd lmZdd lmZdd lmZd d ZddZddZed)ddZeeZed*ddZddZe jdddeeej ddZ!e jdddeeej"ddZ#e jdddeeej$ddZ%eeZ&e&ej%ej$d Z$e jdd!deeej'Z(e&ej)ej'd Z*e(j+dk re(j+,d"d#e(_+e(j+,d$d%e(_+d&d!Z)d'd(Z'e(j+e)_+e*j+e'_+dS)+partial)productN)curry)array_creation_dispatch)Arraynormalize_chunks)meta_from_array)tokenize) blockwise)ArrayChunkShapeDep)funcnamecCst|tjr|}t|ttfs(|f}|dd}|dd}|dd}|dkrf||f||j}t|}t|||d}|pt |dt ||||||}|||||dS)Nnamechunksautodtyper-)shaperkwargsrr) isinstancenpZndarraytolisttuplelistpoprrr r )funcargsrrrrrr3/tmp/pip-unpacked-wheel-dbjnr7gq/dask/array/wrap.py_parse_wrap_argss2     r c Osd|kr |d|dd}}n |d}t|trasz"wrap_func_like..csg|]}qSrr)r,_)rrrr-dsc3s,|]$\}}tfdi|fVqdS)rNr)r,ks)rrrrr gsz!wrap_func_like..)r#) r r(rr rr enumeratezipdictrZastype)rrrxr#rr)rrkeysZshapeskwir0valsZdskr)rrrrrwrap_func_likeRs$r:cKsZ|dkrt||f|}nt||f|}d}|jdk rV|d|ji|j|_d|j|_|S)Nz Blocked variant of %(name)s Follows the signature of %(name)s exactly except that it also features optional keyword arguments ``chunks: int, tuple, or dict`` and ``name: str``. Original signature follows below. rZblocked_)r__doc____name__)Z wrap_funcr func_likerftemplaterrrwrapns  r@rcOs0|dkr dnd}t||f|d|i||S)Nrr"r)rZ broadcast_to)rrr#rrZ null_shaperrr_broadcast_trick_innersrAcCst|}|j|_|j|_|S)a Provide a decorator to wrap common numpy function with a broadcast trick. Dask arrays are currently immutable; thus when we know an array is uniform, we can replace the actual data by a single value and have all elements point to it, thus reducing the size. >>> 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. )rAr;r<)rinnerrrrbroadcast_tricksrCnumpyones)backendrZf8rzerosempty)r=fullz*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_WHITESPACEcOsjt|dkr$tdt|jd|dddkrVt|drJ|j|d<n t||d<t|||d|S)Nr$fill_value must be scalar. Received instead.r)r fill_value) rndim ValueErrortyper<r(hasattrr_full)rrLrrrrrrIs   cOs8t|dkr$tdt|jdt|||d|S)NrrJrK)arL)rrMrNrOr< _full_like)rRrLrrrrr full_likesrT)N)r)- functoolsr itertoolsrrDrZtlzrZdask.array.backendsrZdask.array.corerrZdask.array.utilsr Z dask.baser Zdask.blockwiser r'Z dask.layersr Z dask.utilsr r r+r:r@wrArCZregister_inplaceZ ones_likerEZ zeros_likerGZ empty_likerHZw_likerTrQrIrSr;replacerrrrsv         $