B <ÓFdQ"ã@spdZddlmZmZmZddlZddgZGdd„deƒZ dd „Z ddd „Zdd„Zddd„Z dd„Zdd„ZdS)z¢ Utilities that manipulate strides to achieve desirable effects. An explanation of strides can be found in the "ndarray.rst" file in the NumPy reference guide. é)ÚdivisionÚabsolute_importÚprint_functionNÚbroadcast_toÚbroadcast_arraysc@seZdZdZddd„ZdS)Ú DummyArrayz„Dummy object that just exists to hang __array_interface__ dictionaries and possibly keep alive a reference to a base array. NcCs||_||_dS)N)Ú__array_interface__Úbase)ÚselfÚ interfacer ©rúJ/opt/alt/python37/lib64/python3.7/site-packages/numpy/lib/stride_tricks.pyÚ__init__szDummyArray.__init__)N)Ú__name__Ú __module__Ú__qualname__Ú__doc__rrrrr rsrcCs4t|ƒt|ƒk r0|jt|ƒd}|jr0| |¡|S)N)Útype)rÚviewZ__array_finalize__)Zoriginal_arrayZ new_arrayrrr Ú_maybe_view_as_subclasss rFTcCs~tj|d|d}t|jƒ}|dk r.t|ƒ|d<|dk rBt|ƒ|d<t t||d¡}|j|_t||ƒ}|j j rz|szd|j _ |S)aœ Create a view into the array with the given shape and strides. .. warning:: This function has to be used with extreme care, see notes. Parameters ---------- x : ndarray Array to create a new. shape : sequence of int, optional The shape of the new array. Defaults to ``x.shape``. strides : sequence of int, optional The strides of the new array. Defaults to ``x.strides``. subok : bool, optional .. versionadded:: 1.10 If True, subclasses are preserved. writeable : bool, optional .. versionadded:: 1.12 If set to False, the returned array will always be readonly. Otherwise it will be writable if the original array was. It is advisable to set this to False if possible (see Notes). Returns ------- view : ndarray See also -------- broadcast_to: broadcast an array to a given shape. reshape : reshape an array. Notes ----- ``as_strided`` creates a view into the array given the exact strides and shape. This means it manipulates the internal data structure of ndarray and, if done incorrectly, the array elements can point to invalid memory and can corrupt results or crash your program. It is advisable to always use the original ``x.strides`` when calculating new strides to avoid reliance on a contiguous memory layout. Furthermore, arrays created with this function often contain self overlapping memory, so that two elements are identical. Vectorized write operations on such arrays will typically be unpredictable. They may even give different results for small, large, or transposed arrays. Since writing to these arrays has to be tested and done with great care, you may want to use ``writeable=False`` to avoid accidental write operations. For these reasons it is advisable to avoid ``as_strided`` when possible. F)ÚcopyÚsubokNÚshapeÚstrides)r )ÚnpÚarrayÚdictrÚtupleZasarrayrZdtyperÚflagsÚ writeable)Úxrrrrrrrrrr Ú as_strided&s9 r!c CsÆt |¡rt|ƒn|f}tj|d|d}|s:|jr:tdƒ‚tdd„|DƒƒrTtdƒ‚|o`|jj}|rldgng}|rxdnd }tj |fd ddg||g|d dj d}t||ƒ}|rÂ|jjsÂd|j_|S)NF)rrz/cannot broadcast a non-scalar to a scalar arraycss|]}|dkVqdS)rNr)Ú.0Úsizerrr ú xsz _broadcast_to..z4all elements of broadcast shape must be non-negativeZ reduce_okZ readwriteÚreadonlyZmulti_indexZrefs_okZzerosize_okÚC)rZop_flagsZ itershapeÚorderrT)rÚiterablerrrÚ ValueErrorÚanyrrZnditerZitviewsr) rrrr%Zneeds_writeableZextrasZop_flagÚ broadcastÚresultrrr Ú _broadcast_toss r-cCst|||ddS)aÌBroadcast an array to a new shape. Parameters ---------- array : array_like The array to broadcast. shape : tuple The shape of the desired array. subok : bool, optional If True, then sub-classes will be passed-through, otherwise the returned array will be forced to be a base-class array (default). Returns ------- broadcast : array A readonly view on the original array with the given shape. It is typically not contiguous. Furthermore, more than one element of a broadcasted array may refer to a single memory location. Raises ------ ValueError If the array is not compatible with the new shape according to NumPy's broadcasting rules. Notes ----- .. versionadded:: 1.10.0 Examples -------- >>> x = np.array([1, 2, 3]) >>> np.broadcast_to(x, (3, 3)) array([[1, 2, 3], [1, 2, 3], [1, 2, 3]]) T)rr%)r-)rrrrrr r‡s&cGsb|sdStj|dd…Ž}x@tdt|ƒdƒD],}td|jƒ}tj|f|||d…žŽ}q,W|jS)ztReturns the shape of the arrays that would result from broadcasting the supplied arrays against each other. rNé ér)rr+ÚrangeÚlenrr)ÚargsÚbÚposrrr Ú_broadcast_shape°s r5csn| dd¡‰|r&td | ¡d¡ƒ‚‡fdd„|Dƒ}t|Ž‰t‡fdd„|DƒƒrZ|S‡‡fd d„|DƒS) a± Broadcast any number of arrays against each other. Parameters ---------- `*args` : array_likes The arrays to broadcast. subok : bool, optional If True, then sub-classes will be passed-through, otherwise the returned arrays will be forced to be a base-class array (default). Returns ------- broadcasted : list of arrays These arrays are views on the original arrays. They are typically not contiguous. Furthermore, more than one element of a broadcasted array may refer to a single memory location. If you need to write to the arrays, make copies first. Examples -------- >>> x = np.array([[1,2,3]]) >>> y = np.array([[1],[2],[3]]) >>> np.broadcast_arrays(x, y) [array([[1, 2, 3], [1, 2, 3], [1, 2, 3]]), array([[1, 1, 1], [2, 2, 2], [3, 3, 3]])] Here is a useful idiom for getting contiguous copies instead of non-contiguous views. >>> [np.array(a) for a in np.broadcast_arrays(x, y)] [array([[1, 2, 3], [1, 2, 3], [1, 2, 3]]), array([[1, 1, 1], [2, 2, 2], [3, 3, 3]])] rFz:broadcast_arrays() got an unexpected keyword argument {!r}rcsg|]}tj|dˆd‘qS)F)rr)rr)r"Ú_m)rrr ú ÷sz$broadcast_arrays..c3s|]}|jˆkVqdS)N)r)r"r)rrr r$ûsz#broadcast_arrays..csg|]}t|ˆˆdd‘qS)F)rr%)r-)r"r)rrrr r7s)ÚpopÚ TypeErrorÚformatÚkeysr5Úall)r2Úkwargsr)rrr rÃs0)NNFT)F)rZ __future__rrrZnumpyrÚ__all__Úobjectrrr!r-rr5rrrrr Ús M )