ó ¦–Õ\c@`sÐddlmZmZmZddlmZddlmZddlZ ddl m Z ddl m Z dd l mZdd lmZdd lmZdd lmZd „Zd„Zd„Zd„ZdS(i(tabsolute_importtdivisiontprint_function(tproduct(tmulNi(tArrayi(ttokenize(tflatten(treduce(tHighLevelGraph(tMc `sÐtd„|Dƒƒst‚t|ƒd}t|ƒd}gtt|ƒƒD] }d^qO}gtt|ƒƒD] }d^qt}x1|dks¡|dkr¹||||krë||||<||||<|d8}|d8}q‰n||}||} |dkr"d||<|d8}q‰| dkrEd||<|d8}q‰|| kr‘|d} x:| dkr—tt|| |d!ƒ| kr—| d8} q^Wtt|| |d!ƒ| krÇtdƒ‚nx0t| d|dƒD]}||f||siisShapes not compatiblec3`s|]}ˆ|VqdS(N((R R(tprod(s1lib/python2.7/site-packages/dask/array/reshape.pys 4sc3`s|]}|ˆVqdS(N((R R(tcs(s1lib/python2.7/site-packages/dask/array/reshape.pys Gs(i(i( talltAssertionErrortlentrangetNoneRRt ValueErrortmapt expand_tupleR tcontract_tuple( tinshapetoutshapetinchunkstiitoititresult_inchunkstresult_outchunkstdintdouttilefttchunk_reductiontoleft((RRs1lib/python2.7/site-packages/dask/array/reshape.pytreshape_rechunks`%%            / '$    / $ cC`s¸|dkr|Sg}xw|D]o}|}t||dƒ}x7|d|kru|jt|ƒƒ|t|ƒ8}q?W|r|j|ƒqqWt|ƒt|ƒks®t‚t|ƒS(sÖ >>> expand_tuple((2, 4), 2) (1, 1, 2, 2) >>> expand_tuple((2, 4), 3) (1, 1, 1, 1, 2) >>> expand_tuple((3, 4), 2) (1, 2, 2, 2) >>> expand_tuple((7, 4), 3) (2, 2, 3, 1, 1, 2) ii(tmaxtappendtinttsumRR (tchunkstfactortoutRtxtpart((s1lib/python2.7/site-packages/dask/array/reshape.pyROs  cC`st|ƒ|dkst‚g}d}xL|D]D}||7}||}||}||}|r/|j|ƒq/q/Wt|ƒS(s˜ Return simple chunks tuple such that factor divides all elements Examples -------- >>> contract_tuple((2, 2, 8, 4), 4) (4, 8, 4) i(R+RR)R (R,R-R.tresidualtchunktdivtgood((s1lib/python2.7/site-packages/dask/array/reshape.pyRns      c`sddlm}tt||ƒƒ}g|D]}|dkr,|^q,}t|ƒt|ƒkrðt|ƒt|ƒdkrtdƒ‚nt|ƒdkr²|jdkr²|S||jtt |dƒƒ‰t‡fd†|Dƒƒ}nt j t |j ƒƒrtd|j ƒ‚ntt |dƒ|jkrGtdƒ‚n|j |krZ|Sdt||ƒ}|jdkrtt|jƒƒƒ}itj||f|fdt|ƒ6}td „|Dƒƒ}tj||d |gƒ} t| ||d |jƒSt|j ||jƒ\} } |j| ƒ} tt| jgg| D]} tt| ƒƒ^qKŒƒ}tt|gg| D]} tt| ƒƒ^q‚Œƒ}tt| Œƒ}d „t |||ƒDƒ}tj||d | gƒ} t| || d |jƒS(s² Reshape array to new shape This is a parallelized version of the ``np.reshape`` function with the following limitations: 1. It assumes that the array is stored in `row-major order`_ 2. It only allows for reshapings that collapse or merge dimensions like ``(1, 2, 3, 4) -> (1, 6, 4)`` or ``(64,) -> (4, 4, 4)`` .. _`column-major order`: https://en.wikipedia.org/wiki/ Row-_and_column-major_order When communication is necessary this algorithm depends on the logic within rechunk. It endeavors to keep chunk sizes roughly the same when possible. See Also -------- dask.array.rechunk numpy.reshape i(tsanitize_indexiÿÿÿÿs&can only specify one unknown dimensionc3`s'|]}|dkrˆn|VqdS(iÿÿÿÿN((R ts(t missing_size(s1lib/python2.7/site-packages/dask/array/reshape.pys ¦ss/Array chunk size or shape is unknown. shape: %ss)total size of new array must be unchangedsreshape-ics`s|]}|fVqdS(N((R td((s1lib/python2.7/site-packages/dask/array/reshape.pys ¶st dependenciestdtypecS`s.i|]$\}}}tj||f|“qS((R treshape(R tatbtshape((s1lib/python2.7/site-packages/dask/array/reshape.pys Âs (i(!tslicingR5R RRRtndimtsizeRRtnptisnanR+R>Rt npartitionstnextRt __dask_keys__R R;R tfrom_collectionsRR:R'R,trechunktlistRtnameRtzip(R/R>R5R6t known_sizesRJtkeytdskR,tgraphRt outchunkstx2Rtin_keystout_keystshapes((R7s1lib/python2.7/site-packages/dask/array/reshape.pyR;…s>%!*:7(t __future__RRRt itertoolsRtoperatorRtnumpyRBtcoreRtbaseRRt compatibilityRthighlevelgraphR tutilsR R'RRR;(((s1lib/python2.7/site-packages/dask/array/reshape.pyts  ?