\c@`sddlmZmZmZddlmZmZmZddlm Z ddl m Z m Z ddl mZmZddlZddlmZmZdd lmZdd lmZdd lmZd d lmZd dlmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%d dl&m'Z'm(Z(m)Z)m*Z*d dl+m,Z,dddZ.dddZ/dddZ0dddZ1de2e3dddZ4dZ5eej6dZ6e7ddZ8dde9dZ:eej;dZ;eej<ddd dZ<ddZ=ddZ>d Z?eej@dddd!Z@eejAdd"ZAeejBd#ZBd$ZCd%ZDd&ZEd'ZFd(ZGd)ZHd*ZId+ZJeejKd,ZKdS(-i(tabsolute_importtdivisiontprint_function(tpartialtwrapstreduce(tproduct(taddtgetitem(tIntegraltNumberN(t accumulatetsliding_windowi(tHighLevelGraph(ttokenize(tSequencei(tchunk(tArraytasarraytnormalize_chunkststackt concatenatetblockt broadcast_totbroadcast_arrays(temptytonestzerostfull(t AxisErrorcC`sIt|dt}t|jd|p*|jd|dk r?|n|jS(sz Return a new array with the same shape and type as a given array. Parameters ---------- a : array_like The shape and data-type of `a` define these same attributes of the returned array. dtype : data-type, optional Overrides the data type of the result. chunks : sequence of ints The number of samples on each block. Note that the last block will have fewer samples if ``len(array) % chunks != 0``. Returns ------- out : ndarray Array of uninitialized (arbitrary) data with the same shape and type as `a`. See Also -------- ones_like : Return an array of ones with shape and type of input. zeros_like : Return an array of zeros with shape and type of input. empty : Return a new uninitialized array. ones : Return a new array setting values to one. zeros : Return a new array setting values to zero. Notes ----- This function does *not* initialize the returned array; to do that use `zeros_like` or `ones_like` instead. It may be marginally faster than the functions that do set the array values. tnametdtypetchunksN(RtFalseRtshapeRtNoneR (taRR ((s2lib/python2.7/site-packages/dask/array/creation.pyt empty_likes$cC`sIt|dt}t|jd|p*|jd|dk r?|n|jS(s Return an array of ones with the same shape and type as a given array. Parameters ---------- a : array_like The shape and data-type of `a` define these same attributes of the returned array. dtype : data-type, optional Overrides the data type of the result. chunks : sequence of ints The number of samples on each block. Note that the last block will have fewer samples if ``len(array) % chunks != 0``. Returns ------- out : ndarray Array of ones with the same shape and type as `a`. See Also -------- zeros_like : Return an array of zeros with shape and type of input. empty_like : Return an empty array with shape and type of input. zeros : Return a new array setting values to zero. ones : Return a new array setting values to one. empty : Return a new uninitialized array. RRR N(RR!RR"RR#R (R$RR ((s2lib/python2.7/site-packages/dask/array/creation.pyt ones_likeAscC`sIt|dt}t|jd|p*|jd|dk r?|n|jS(s Return an array of zeros with the same shape and type as a given array. Parameters ---------- a : array_like The shape and data-type of `a` define these same attributes of the returned array. dtype : data-type, optional Overrides the data type of the result. chunks : sequence of ints The number of samples on each block. Note that the last block will have fewer samples if ``len(array) % chunks != 0``. Returns ------- out : ndarray Array of zeros with the same shape and type as `a`. See Also -------- ones_like : Return an array of ones with shape and type of input. empty_like : Return an empty array with shape and type of input. zeros : Return a new array setting values to zero. ones : Return a new array setting values to one. empty : Return a new uninitialized array. RRR N(RR!RR"RR#R (R$RR ((s2lib/python2.7/site-packages/dask/array/creation.pyt zeros_likeescC`sLt|dt}t|j|d|p-|jd|dk rB|n|jS(s Return a full array with the same shape and type as a given array. Parameters ---------- a : array_like The shape and data-type of `a` define these same attributes of the returned array. fill_value : scalar Fill value. dtype : data-type, optional Overrides the data type of the result. chunks : sequence of ints The number of samples on each block. Note that the last block will have fewer samples if ``len(array) % chunks != 0``. Returns ------- out : ndarray Array of `fill_value` with the same shape and type as `a`. See Also -------- zeros_like : Return an array of zeros with shape and type of input. ones_like : Return an array of ones with shape and type of input. empty_like : Return an empty array with shape and type of input. zeros : Return a new array setting values to zero. ones : Return a new array setting values to one. empty : Return a new uninitialized array. full : Fill a new array. RRR N(RR!RR"RR#R (R$t fill_valueRR ((s2lib/python2.7/site-packages/dask/array/creation.pyt full_likes !i2tautocC`sot|}|dkr3tjdddj}nt||fd|}||}|re|dn|}t||} dt||||||f} i} |} xt|dD]u\} }|r|dn|}| || }t tjd|d|| ||f}| | |} || | | f((s2lib/python2.7/site-packages/dask/array/creation.pyREs>"      & +c O`s|jdd}t|jdt}d|krEtdn|rZtdn|d krutdng|D]}t|^q|}g|D]}|j^q}|dkrt|d kr|d |d |d <|d s t dependenciess3Extracting diagonals from non-square chunked arrays(RRaR-tndarraytndimR"RmtminRRRRRAtformatttypeR R0t __dask_keys__R tfrom_collectionsRQR( tvR R8tgrapht chunks_1dtblocksR:tmRktntkey((Rs2lib/python2.7/site-packages/dask/array/creation.pyRms> "   #   !c `sdt|||}jdkr7tdnd}||dj}||dj}||krtdnt||kr||}}| }nd}g}g}d gttjj|d } d gttjj|d } xt| j|D]x\} } |j gx\t| j|D]D\} }|| | }|j || |||d j |q^Wq.Wi}t t jt ||g}t j|}t j|}xt fd |DD]}xtt t |t |D]\}\}}tt|| j|}tt|||d !||}tt||d |}|||}tj||||f||f||f(sic3`s|]}j|VqdS(N(R"(RnR:(R$(s2lib/python2.7/site-packages/dask/array/creation.pys 0sc3`s|]}j|VqdS(N(R (RnR:(R$(s2lib/python2.7/site-packages/dask/array/creation.pys 4sRpR"R R(RRrRRRtlistR-tcumsumR tzipRWtsetRTR@RR0RRRvtdiagonalRVR"R RwRR(R$RRRRRRt diag_chunkst chunk_offsetstcum1tcum2tco1tc1tco2tc2ReR8tidx_settn1tn2tidxR:ti1ti2ttskt left_shapet right_shapeR"t left_chunksR Ry((R$s2lib/python2.7/site-packages/dask/array/creation.pyRsT       ''# #%#4 !2#  c C`s|jdkrtdn|jdd|jddkrUd}t|nt|jd}t|jd}|jdd}t||}d|}i}x t|D]} xt|D]} || | d|kr$tj|jd| |jd| ff||| | f 0` is above. Returns ------- triu : ndarray, shape (M, N) Upper triangle of `m`, of same shape and data-type as `m`. See Also -------- tril : lower triangle of an array isinput must be 2 dimensionaliisJchunks must be a square. Use .rechunk method to change the size of chunks.striu-RpR"R R(RrRRR RQR@RRTR-RttriuRR RwRR"R( R|ReRtrdimthdimRRfRR8R:RkRy((s2lib/python2.7/site-packages/dask/array/creation.pyR<s(" ;4:'c C`s|jdkrtdntt|jd|jddks]d}t|nt|jd}t|jd}|jdd}t||}d|}i}x t|D]} xt|D]} || | d|kr|j| | f||| | f 0` is above. Returns ------- tril : ndarray, shape (M, M) Lower triangle of `m`, of same shape and data-type as `m`. See Also -------- triu : upper triangle of an array isinput must be 2 dimensionaliisqAll chunks must be a square matrix to perform lu decomposition. Use .rechunk method to change the size of chunks.stril-RpR"R R(RrRRR@RR RRTRR-ttrilRR RwRR"R( R|ReRRRRRfRR8R:RkRy((s2lib/python2.7/site-packages/dask/array/creation.pyRls(* "4:@c`s+fd}tj||d||S(Nc`s%ttt|}||S(N(RtmapR(RHRItargs2(tfuncR(s2lib/python2.7/site-packages/dask/array/creation.pyt offset_funcsR(R-t fromfunction(RR"RRt func_kwargsR((RRs2lib/python2.7/site-packages/dask/array/creation.pyt_np_fromfunctionscK`s-t||d|}dt|||||}tt|gg|D]}tt|^qD}g|D]#}tttd|d ^qo}tt|} tt|} |pt}gt | | D]$\} } t || || |f^q} t t || }t |||d|S(NRs fromfunction-ii(i( RRRRRTR@R RR/RRtdictR(RR R"RRIRtbdRGtaggdimstoffsetstshapesRtshptvaluesR8((s2lib/python2.7/site-packages/dask/array/creation.pyRs70 7c C`s|dkr3|jdkr$d}q3tdnt|tsQtdn|j |koldknr||j7}n6d|ko|jdknstd|n|dkr|Stjd |j|}g}xt d|D]o\}}tj |||j d}xBt d|D]1\}} || kr1|j t || q1q1WqWt ddd} g|D]0} | f|| f| f|j|d^q}g|D]} || ^q} g}x| D]}t|j}t||dkst||d|f||siiics`s|]}t|tVqdS(N(RaR (RnR((s2lib/python2.7/site-packages/dask/array/creation.pys sc3`s#|]}ddfVqdS(iiN((Rnt_(t pad_value(s2lib/python2.7/site-packages/dask/array/creation.pys scs`s|]}t|tVqdS(N(RaR(RnR((s2lib/python2.7/site-packages/dask/array/creation.pys scs`s!|]}t|dkVqdS(iN(R@(RnR((s2lib/python2.7/site-packages/dask/array/creation.pys scs`s%|]}td|DVqdS(cs`s|]}t|tVqdS(N(RaR (Rntw((s2lib/python2.7/site-packages/dask/array/creation.pys sN(tall(RnR((s2lib/python2.7/site-packages/dask/array/creation.pys scs`s#|]}|d|dfVqdS(iiN((RnR((s2lib/python2.7/site-packages/dask/array/creation.pys ss6`pad_value` must be composed of integral typed values.( RaR RrRRR@RVRTRA(tarrayR((Rs2lib/python2.7/site-packages/dask/array/creation.pytexpand_pad_values(!% c C`st|jt|jg}t|jt|jg}xT|D]L}xCtdD]5}||||||<|||f||| Helper function to find the linear ramp for a chunk. iRN( RR"RVR-RRtndindexRUR#R.( R1R2R3tdimR7tnum1R"RRR:Rk((s2lib/python2.7/site-packages/dask/array/creation.pytlinear_ramp_chunk!s     8c`stfd|D}}xjtjD]Y}t|||f\}}||g}|dkr|d|} g| D]} t| j|j^q} gt| ||D]!\} } } t| | | ^q}n|dkrh|jt dg|jt dgg} t ddd| d|@stconstantitedget linear_rampiiR RRR7iR(RRN(RVRTRrRRtastypeRRRRUR#R0RRR(RRtmodeRHRRRRt pad_arraystconstant_valuesRRxR[t pad_slicestslR$t end_valuesR:tevR((Rs2lib/python2.7/site-packages/dask/array/creation.pytpad_edge9s4  +: , : v'c G`s:|d kr'd|kr'tdntj|jddt}xtj|jD]}g}g}xft||j|D]O\}} } |dkr| ddd } n|d kr |dkr|jt d | d d dq|jt d| d dn{|d krs|dkrO|jt | | d d | d dq|jt | | d ddn|jt d|d kr|dkr|jt ddd q~|jt dq~Wt |}t |}|dkrt d |D}n|||||s(RR(i(RR(RQR-RRrtobjectRR"RRWRUR#RVRttolist( RRRRHRRtselecttorientR:R[R((s2lib/python2.7/site-packages/dask/array/creation.pyt pad_reusers:  %   $   ,$   cG`s|dkrtdnt||d}tj|jd dt}x+tj|jD]}g}g}g} g} x(tt ||j|j ||D]\} \} } }}}| dkr|j | |j t d|dd| j |d| j |dq| dkru|j | |j t | |ddd| j |d| j |dq|j t d| j | | j |qWt|}t|}t| } t| } ||}|dkr|jd|d t}nN|d kr.|jd|d t}n'|d krU|jd|d t}nt|| d | }|||sV :"+$$*  D M'14( B 0 0   -    9 5 < !