3
���h@# � @ s� d Z ddlZddlmZ ddgZG dd� d�Zdd � Zdd d
�Zdd� Z d dd�Z
ee
dd�d dd��Z dd� Z dd�dd�Z
ee
dd�d
d�dd��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.
� N)�array_function_dispatch� broadcast_to�broadcast_arraysc @ s e Zd ZdZddd�ZdS )�
DummyArrayz�Dummy object that just exists to hang __array_interface__ dictionaries
and possibly keep alive a reference to a base array.
Nc C s || _ || _d S )N)�__array_interface__�base)�self� interfacer � r
�8/tmp/pip-build-5_djhm0z/numpy/numpy/lib/stride_tricks.py�__init__ s zDummyArray.__init__)N)�__name__�
__module__� __qualname__�__doc__r r
r
r
r r s r c C s4 t | �t |�k r0|jt | �d�}|jr0|j| � |S )N)�type)r �viewZ__array_finalize__)Zoriginal_arrayZ new_arrayr
r
r �_maybe_view_as_subclass s
r FTc C s� t j| d|d�} t| j�}|dk r.t|�|d< |dk rBt|�|d< t jt|| d��}| j|_t| |�}|j j
r|| r|d|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 �tupleZasarrayr Zdtyper �flags� writeable)�xr r r r r r r r
r
r �
as_strided% s 9
r c
C s� t j|�rt|�n|f}t j| d|d�} | r<| jr<td��tdd� |D ��rVtd��g }t j| fddd g| d
g|d d �}|� |jd
}W d Q R X t | |�}| r�| j
j r�d|j
_ d|j
_
|S )NF)r r z/cannot broadcast a non-scalar to a scalar arrayc s s | ]}|d k V qdS )r Nr
)�.0�sizer
r
r � <genexpr>w s z _broadcast_to.<locals>.<genexpr>z4all elements of broadcast shape must be non-negativeZ multi_indexZrefs_okZ zerosize_ok�readonly�C)r Zop_flagsZ itershape�orderr T)r �iterabler r r �
ValueError�anyZnditerZitviewsr r Z_writeable_no_warnr Z_warn_on_write)r r r r# �extras�it� broadcast�resultr
r
r �
_broadcast_tor s"
r- c C s | fS )Nr
)r r r r
r
r �_broadcast_to_dispatcher� s r. �numpy)�modulec C s t | ||dd�S )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)r r# )r- )r r r r
r
r r � s 'c G sZ t j| dd� � }x@tdt| �d�D ],}td|j�}t j|f| ||d � �� }q$W |jS )ztReturns the shape of the arrays that would result from broadcasting the
supplied arrays against each other.
N� � r )r r+ �range�lenr r )�args�b�posr
r
r �_broadcast_shape� s
r8 )r c G s |S )Nr
)r r5 r
r
r � _broadcast_arrays_dispatcher� s r9 c sH �fdd�|D �}t |� � t� fdd�|D ��r4|S � �fdd�|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. While you can set the
``writable`` flag True, writing to a single output value may end up
changing more than one location in the output array.
.. deprecated:: 1.17
The output is currently marked so that if written to, a deprecation
warning will be emitted. A future version will set the
``writable`` flag False so writing to it will raise an error.
Examples
--------
>>> x = np.array([[1,2,3]])
>>> y = np.array([[4],[5]])
>>> np.broadcast_arrays(x, y)
[array([[1, 2, 3],
[1, 2, 3]]), array([[4, 4, 4],
[5, 5, 5]])]
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]]), array([[4, 4, 4],
[5, 5, 5]])]
c s g | ]}t j|d � d��qS )F)r r )r r )r �_m)r r
r �
<listcomp> s z$broadcast_arrays.<locals>.<listcomp>c 3 s | ]}|j � kV qd S )N)r )r r )r r
r r" s z#broadcast_arrays.<locals>.<genexpr>c s g | ]}t |� �d d��qS )F)r r# )r- )r r )r r r
r r; s )r8 �all)r r5 r
)r r r r � s 4 )NNFT)N)F)r r/ r Znumpy.core.overridesr �__all__r r r r- r. r r8 r9 r r
r
r
r �<module> s
M
)
|