U
|
O ±d8 ã @ sÎ d Z ddlmZ ddlZddlmZ ddlmZ ddlm Z m
|
Z
|
m Z m Z m Z mZmZmZ ddlmZmZmZmZmZmZ ddlZer¨dd lmZmZmZm Z ddl mZ ddl Z dd
|
l m!Z! G d d d Z"dS ) al
|
Wrapper class around the ndarray object for the array API standard.
|
|
The array API standard defines some behaviors differently than ndarray, in
|
particular, type promotion rules are different (the standard has no
|
value-based casting). The standard also specifies a more limited subset of
|
array methods and functionalities than are implemented on ndarray. Since the
|
goal of the array_api namespace is to be a minimal implementation of the array
|
API standard, we need to define a separate wrapper class for the array_api
|
namespace.
|
|
The standard compliant class is only a wrapper class. It is *not* a subclass
|
of ndarray.
|
é )Ú annotationsN)ÚIntEnumé )Úasarray)Ú _all_dtypesÚ_boolean_dtypesÚ_integer_dtypesÚ_integer_or_boolean_dtypesÚ_floating_dtypesÚ_numeric_dtypesÚ _result_typeÚ_dtype_categories)Ú TYPE_CHECKINGÚOptionalÚTupleÚUnionÚAnyÚ SupportsIndex)r Ú PyCapsuleÚDeviceÚDtype)Ú array_apic s@ e Zd ZU dZded< e fddZdd Zd dd d
|
d Zd dd d d Z d½dddddZ
|
dddd dddZ dd Z e ddd d Zd d Zd d d d d!Zd d"d d#d$d%Zd d&d d#d'd(Zdd)d d*d+d,d-d.Zd d/d d0d1Zdd2d d3d4d5d6d7Zd d8d d9d:Zd d;d d#d<d=Zd d>d d?d@Zd d"d d#dAdBZd d"d d#dCdDZd dEd dFdGdHZd d"d d#dIdJZ d dKd dLdMZ d dKd dNdOZ d d d dPdQZ d d"d d#dRdSZ d dTd d#dUdVZ!d d"d d#dWdXZ"d d d d#dYdZZ#d d"d d#d[d\Z$d d"d d#d]d^Z%d d;d d#d_d`Z&d d d dadbZ'd d&d d#dcddZ(d d d dedfZ)d d"d d#dgdhZ*d dTd d#didjZ+dEd;d3dkdldmZ,d d"d d#dndoZ-d dpd d#dqdrZ.d d&d d#dsdtZ/d d"d d#dudvZ0d d"d d#dwdxZ1d d&d d#dydzZ2d d&d d#d{d|Z3d d"d d#d}d~Z4d d"d d#ddZ5d dTd d#ddZ6d dTd d#ddZ7d d d d#d
dZ8d d d d#ddZ9d d"d d#ddZ:d d"d d#ddZ;d d"d d#ddZ<d d"d d#ddZ=d d&d d#ddZ>d d&d d#ddZ?d d"d d#ddZ@d d"d d#ddZAd dTd d#ddZBd dTd d#ddZCd d"d d#ddZDd d"d d#dd ZEd dpd d#d¡d¢ZFd dpd d#d£d¤ZGd d&d d#d¥d¦ZHd d&d d#d§d¨ZId¾d3d d©d dªd«d¬ZJeKddd®d¯ZLeKd©dd°d±ZMeKd dd²d³ZNeKdKdd´dµZOeKd¶dd·d¸ZPeKdKdd¹dºZQeKd dd»d¼ZR ZSS )¿ÚArraya)
|
n-d array object for the array API namespace.
|
|
See the docstring of :py:obj:`np.ndarray <numpy.ndarray>` for more
|
information.
|
|
This is a wrapper around numpy.ndarray that restricts the usage to only
|
those things that are required by the array API namespace. Note,
|
attributes on this object that start with a single underscore are not part
|
of the API specification and should only be used internally. This object
|
should not be constructed directly. Rather, use one of the creation
|
functions, such as asarray().
|
|
z
|
np.ndarrayÚ_arrayc sH t | ¡}t|tjr"t |¡}|jtkr>td|j d||_ |S )a
|
This is a private method for initializing the array API Array
|
object.
|
|
Functions outside of the array_api submodule should not use this
|
method. Use one of the creation functions instead, such as
|
``asarray``.
|
|
z4The array_api namespace does not support the dtype 'ú')
|
ÚsuperÚ__new__Ú
|
isinstanceÚnpZgenericr Údtyper Ú TypeErrorr )ÚclsÚxÚobj©Ú __class__© úTd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/array_api/_array_object.pyÚ_new? s
|
|
ÿz
|
Array._newc O s t dd S )Nz{The array_api Array object should not be instantiated directly. Use an array creation function, such as asarray(), instead.)r )r! ÚargsÚkwargsr& r& r' r W s ÿz Array.__new__Ústr)ÚselfÚreturnc C s | j ¡ dd¡S )z1
|
Performs the operation __str__.
|
Úarrayr )r Ú__str__Úreplace©r, r& r& r' r/ _ s z Array.__str__c C sN d| j j d}d| jkr*d}t| j}nd}tj| jd||d}|| | S )z2
|
Performs the operation __repr__.
|
z, dtype=ú)r zempty(zArray(z, )Ú separatorÚprefixÚsuffix)r ÚnameÚshaper+ r Z array2stringr )r, r5 r4 Zmidr& r& r' Ú__repr__e s
|
zArray.__repr__NzNone | np.dtype[Any]znpt.NDArray[Any])r r- c C s t j| j|dS )zØ
|
Warning: this method is NOT part of the array API spec. Implementers
|
of other libraries need not include it, and users should not assume it
|
will be present in other implementations.
|
|
)r )r r r )r, r r& r& r' Ú __array__t s zArray.__array__zbool | int | float | Array)ÚotherÚdtype_categoryÚopr- c C s´ | j t| kr"td| d| t|tttfr>| |¡}n2t|trl|j t| krptd| d| nt S t
|
| j |j }| d¡r°|| j kr°td| d| j d|j |S )z÷
|
Helper function for operators to only allow specific input dtypes
|
|
Use like
|
|
other = self._check_allowed_dtypes(other, 'numeric', '__add__')
|
if other is NotImplemented:
|
return other
|
zOnly z dtypes are allowed in Z__izCannot perform z with dtypes ú and ) r r r r ÚintÚfloatÚboolÚ_promote_scalarr ÚNotImplementedr Ú
|
startswith)r, r: r; r< Z res_dtyper& r& r' Ú_check_allowed_dtypes s
|
|
|
ÿzArray._check_allowed_dtypesc C sv t |tr | jtkrbtdnDt |tr<| jtkrbtdn&t |trZ| jtkrbtdntdt t
|
|| j¡¡S )aR
|
Returns a promoted version of a Python scalar appropriate for use with
|
operations on self.
|
|
This may raise an OverflowError in cases where the scalar is an
|
integer that is too large to fit in a NumPy integer dtype, or
|
TypeError when the scalar type is incompatible with the dtype of self.
|
z9Python bool scalars can only be promoted with bool arraysz6Python int scalars cannot be promoted with bool arrayszEPython float scalars can only be promoted with floating-point arrays.z 'scalar' must be a Python scalar) r r@ r r r r> r? r
|
r r( r r. )r, Zscalarr& r& r' rA « s"
|
|
ÿ
|
|
ÿ
|
|
ÿ zArray._promote_scalarzTuple[Array, Array])r- c C sR | j dkr&|j dkr&t | jd ¡} n$|j dkrJ| j dkrJt |jd ¡}| |fS )aâ
|
Normalize inputs to two arg functions to fix type promotion rules
|
|
NumPy deviates from the spec type promotion rules in cases where one
|
argument is 0-dimensional and the other is not. For example:
|
|
>>> import numpy as np
|
>>> a = np.array([1.0], dtype=np.float32)
|
>>> b = np.array(1.0, dtype=np.float64)
|
>>> np.add(a, b) # The spec says this should be float64
|
array([2.], dtype=float32)
|
|
To fix this, we add a dimension to the 0-dimension array before passing it
|
through. This works because a dimension would be added anyway from
|
broadcasting, so the resulting shape is the same, but this prevents NumPy
|
from not promoting the dtype.
|
r N)Úndimr r( r )Úx1Zx2r& r& r' Ú_normalize_two_argsÑ s
|
zArray._normalize_two_argsc C s t |tr|n|f}|D ]b}t |ts`t |tst |ts|tks|dkst |tst |tjst d| dt
|
|dqg }g }d}d}|D ]b}|dk r| |¡ t |ts¼t |tjrÖ|j t krÊd}| |¡ q|tkrè|d7 }q| |¡ qt|}|dkr
|
dS |dkr>|s>|| jk r>t d | jd
|
| d |dkrP| j} n~d}
|
t|D ]8\} }t |ts\t |tjs\|tkr\| }
|
qq\|
|
dk s¤t| j||
|
} | jd|
|
| j| d
} t|| D ]º\}} t |trú| dkrþd }nd | d| d d}|jdk rzt |j¡}W n tk
|
rF Y n@X | | krb| ksn t d| d|d| d| d |jdk rzt |j¡}W n tk
|
r¸ Y n@X | | krÔ| ksn t d| d|d| d| d nt |trv|j t krLt|dkrLt |ts0tt d| dt|dn(|j tkr|jdkrt d| dn t |trØt d| dqØdS )aÄ
|
Validate an index according to the array API.
|
|
The array API specification only requires a subset of indices that are
|
supported by NumPy. This function will reject any index that is
|
allowed by NumPy but not required by the array API specification. We
|
always raise ``IndexError`` on such indices (the spec does not require
|
any specific behavior on them, but this makes the NumPy array API
|
namespace a minimal implementation of the spec). See
|
https://data-apis.org/array-api/latest/API_specification/indexing.html
|
for the full list of required indexing behavior
|
|
This function raises IndexError if the index ``key`` is invalid. It
|
only raises ``IndexError`` on indices that are not already rejected by
|
NumPy, as NumPy will already raise the appropriate error on such
|
indices. ``shape`` may be None, in which case, only cases that are
|
independent of the array shape are checked.
|
|
The following cases are allowed by NumPy, but not specified by the array
|
API specification:
|
|
- Indices to not include an implicit ellipsis at the end. That is,
|
every axis of an array must be explicitly indexed or an ellipsis
|
included. This behaviour is sometimes referred to as flat indexing.
|
|
- The start and stop of a slice may not be out of bounds. In
|
particular, for a slice ``i:j:k`` on an axis of size ``n``, only the
|
following are allowed:
|
|
- ``i`` or ``j`` omitted (``None``).
|
- ``-n <= i <= max(0, n - 1)``.
|
- For ``k > 0`` or ``k`` omitted (``None``), ``-n <= j <= n``.
|
- For ``k < 0``, ``-n - 1 <= j <= max(0, n - 1)``.
|
|
- Boolean array indices are not allowed as part of a larger tuple
|
index.
|
|
- Integer array indices are not allowed (with the exception of 0-D
|
arrays, which are treated the same as scalars).
|
|
Additionally, it should be noted that indices that would return a
|
scalar in NumPy will return a 0-D array. Array scalars are not allowed
|
in the specification, only 0-D arrays. This is done in the
|
``Array._new`` constructor, not this function.
|
|
NzSingle-axes index z has type(i)=z, but only integers, slices (:), ellipsis (...), newaxis (None), zero-dimensional integer arrays and boolean arrays are specified in the Array API.r FTr z
|
self.ndim=z*, but the multi-axes index only specifies zê dimensions. If this was intentional, add a trailing ellipsis (...) which expands into as many slices (:) as necessary - this is what np.ndarray arrays implicitly do, but such flat indexing behaviour is not specified in the Array API.z 0 (or None)z between -r= z
|
(or None)zSlice z contains start=z, but should be z for an axis of size z: (out-of-bounds starts are not specified in the Array API)z contains stop=z9 (out-of-bounds stops are not specified in the Array API)z! is a boolean array and len(key)=zR, but masking is only specified in the Array API when the array is the sole index.zj is a non-zero-dimensional integer array, but advanced integer indexing is not specified in the Array API.zI is a tuple, but nested tuple indices are not specified in the Array API.)r Útupler@ r ÚsliceÚEllipsisr r ZndarrayÚ
|
IndexErrorÚtypeÚappendr r ÚlenrE r7 Ú enumerateÚAssertionErrorÚzipÚstartÚoperatorÚindexr Ústopr )r, ÚkeyÚ_keyÚiZnonexpanding_keyZ single_axesZ
|
n_ellipsisZ key_has_maskZ n_single_axesZ indexed_shapeZellipsis_startÚposZ ellipsis_endZsideZf_rangerR rU r& r& r' Ú_validate_indexö s² /
|
ÿþýüû
|
úÿ
|
|
|
|
|
ÿ
|
|
|
ÿ
|
ÿ ÿ ÿ
|
ÿ
|
ÿzArray._validate_indexc C s( | j tkrtd| j ¡ }| j |¡S )z1
|
Performs the operation __abs__.
|
z*Only numeric dtypes are allowed in __abs__)r r r r Ú__abs__r% r( ©r, Úresr& r& r' r[ s
|
|
z Array.__abs__zUnion[int, float, Array])r, r: r- c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z1
|
Performs the operation __add__.
|
ÚnumericÚ__add__)rD rB rG r r_ r% r( ©r, r: r] r& r& r' r_ ¥ s z Array.__add__zUnion[int, bool, Array]c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z1
|
Performs the operation __and__.
|
úinteger or booleanÚ__and__)rD rB rG r rb r% r( r` r& r& r' rb ° s z Array.__and__)Ú api_versionz Optional[str]ztypes.ModuleType)r, rc r- c C s$ |d k r | d¡s td|tS )Nz2021.z Unrecognized array API version: )rC Ú
|
ValueErrorr )r, rc r& r& r' Ú__array_namespace__» s zArray.__array_namespace__r@ c C s4 | j jdkrtd| jtkr&td| j ¡ }|S )z2
|
Performs the operation __bool__.
|
r z0bool is only allowed on arrays with 0 dimensionsz&bool is only allowed on boolean arrays)r rE r r r rd Ú__bool__r\ r& r& r' rf  s
|
|
zArray.__bool__©ÚstreamÚNoner )r, rh r- c C s | j j|dS )z4
|
Performs the operation __dlpack__.
|
rg )r Ú
|
__dlpack__)r, rh r& r& r' rj Î s zArray.__dlpack__zTuple[IntEnum, int]c C s
|
| j ¡ S )z;
|
Performs the operation __dlpack_device__.
|
)r Ú__dlpack_device__r1 r& r& r' rk Ô s zArray.__dlpack_device__z Union[int, float, bool, Array]c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z0
|
Performs the operation __eq__.
|
ÚallÚ__eq__)rD rB rG r rm r% r( r` r& r& r' rm Û s z Array.__eq__r? c C s4 | j jdkrtd| jtkr&td| j ¡ }|S )z3
|
Performs the operation __float__.
|
r z1float is only allowed on arrays with 0 dimensionsz.float is only allowed on floating-point arrays)r rE r r r
|
rd Ú __float__r\ r& r& r' rn è s
|
|
zArray.__float__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z6
|
Performs the operation __floordiv__.
|
r^ Ú __floordiv__)rD rB rG r ro r% r( r` r& r& r' ro ô s zArray.__floordiv__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z0
|
Performs the operation __ge__.
|
r^ Ú__ge__)rD rB rG r rp r% r( r` r& r& r' rp ÿ s z Array.__ge__zKUnion[int, slice, ellipsis, Tuple[Union[int, slice, ellipsis], ...], Array])r, rV r- c C s0 | |¡ t|tr|j}| j |¡}| |¡S )z5
|
Performs the operation __getitem__.
|
)rZ r r r Ú __getitem__r( )r, rV r] r& r& r' rq
|
s
|
|
|
zArray.__getitem__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z0
|
Performs the operation __gt__.
|
r^ Ú__gt__)rD rB rG r rr r% r( r` r& r& r' rr s z Array.__gt__r> c C s4 | j jdkrtd| jtkr&td| j ¡ }|S )z1
|
Performs the operation __int__.
|
r z/int is only allowed on arrays with 0 dimensionsz%int is only allowed on integer arrays)r rE r r r rd Ú__int__r\ r& r& r' rs ( s
|
|
z Array.__int__c C s | j ¡ }|S )z3
|
Performs the operation __index__.
|
)r Ú __index__r\ r& r& r' rt 4 s
|
zArray.__index__c C s( | j tkrtd| j ¡ }| j |¡S )z4
|
Performs the operation __invert__.
|
z8Only integer or boolean dtypes are allowed in __invert__)r r r r Ú
|
__invert__r% r( r\ r& r& r' ru ; s
|
|
zArray.__invert__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z0
|
Performs the operation __le__.
|
r^ Ú__le__)rD rB rG r rv r% r( r` r& r& r' rv D s z Array.__le__zUnion[int, Array]c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z4
|
Performs the operation __lshift__.
|
ÚintegerÚ
|
__lshift__)rD rB rG r rx r% r( r` r& r& r' rx O s zArray.__lshift__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z0
|
Performs the operation __lt__.
|
r^ Ú__lt__)rD rB rG r ry r% r( r` r& r& r' ry Z s z Array.__lt__c C s4 | |dd¡}|tkr|S | j |j¡}| j |¡S )z4
|
Performs the operation __matmul__.
|
r^ Ú
|
__matmul__)rD rB r rz r% r( r` r& r& r' rz e s
|
zArray.__matmul__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z1
|
Performs the operation __mod__.
|
r^ Ú__mod__)rD rB rG r r{ r% r( r` r& r& r' r{ q s z Array.__mod__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z1
|
Performs the operation __mul__.
|
r^ Ú__mul__)rD rB rG r r| r% r( r` r& r& r' r| | s z Array.__mul__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z0
|
Performs the operation __ne__.
|
rl Ú__ne__)rD rB rG r r} r% r( r` r& r& r' r} s z Array.__ne__c C s( | j tkrtd| j ¡ }| j |¡S )z1
|
Performs the operation __neg__.
|
z*Only numeric dtypes are allowed in __neg__)r r r r Ú__neg__r% r( r\ r& r& r' r~ s
|
|
z Array.__neg__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z0
|
Performs the operation __or__.
|
ra Ú__or__)rD rB rG r r r% r( r` r& r& r' r s z Array.__or__c C s( | j tkrtd| j ¡ }| j |¡S )z1
|
Performs the operation __pos__.
|
z*Only numeric dtypes are allowed in __pos__)r r r r Ú__pos__r% r( r\ r& r& r' r ¦ s
|
|
z Array.__pos__c C s0 ddl m} | |dd¡}|tkr&|S || |S )z1
|
Performs the operation __pow__.
|
r ©Úpowr^ Ú__pow__©Z_elementwise_functionsr rD rB ©r, r: r r& r& r' r ¯ s
|
z Array.__pow__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z4
|
Performs the operation __rshift__.
|
rw Ú
|
__rshift__)rD rB rG r r r% r( r` r& r& r' r ¼ s zArray.__rshift__)rV Úvaluer- c C s2 | |¡ t|tr|j}| j |t|j¡ dS )z5
|
Performs the operation __setitem__.
|
N)rZ r r r Ú __setitem__r )r, rV r r& r& r' r Ç s
|
|
zArray.__setitem__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z1
|
Performs the operation __sub__.
|
r^ Ú__sub__)rD rB rG r r r% r( r` r& r& r' r Ú s z Array.__sub__zUnion[float, Array]c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z5
|
Performs the operation __truediv__.
|
úfloating-pointÚ __truediv__)rD rB rG r r r% r( r` r& r& r' r ç s zArray.__truediv__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z1
|
Performs the operation __xor__.
|
ra Ú__xor__)rD rB rG r r r% r( r` r& r& r' r ò s z Array.__xor__c C s, | |dd¡}|tkr|S | j |j¡ | S )z2
|
Performs the operation __iadd__.
|
r^ Ú__iadd__)rD rB r r ©r, r: r& r& r' r ý s
|
zArray.__iadd__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z2
|
Performs the operation __radd__.
|
r^ Ú__radd__)rD rB rG r r r% r( r` r& r& r' r s zArray.__radd__c C s, | |dd¡}|tkr|S | j |j¡ | S )z2
|
Performs the operation __iand__.
|
ra Ú__iand__)rD rB r r r r& r& r' r s
|
zArray.__iand__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z2
|
Performs the operation __rand__.
|
ra Ú__rand__)rD rB rG r r r% r( r` r& r& r' r s zArray.__rand__c C s, | |dd¡}|tkr|S | j |j¡ | S )z7
|
Performs the operation __ifloordiv__.
|
r^ Ú __ifloordiv__)rD rB r r r r& r& r' r ' s
|
zArray.__ifloordiv__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z7
|
Performs the operation __rfloordiv__.
|
r^ Ú __rfloordiv__)rD rB rG r r r% r( r` r& r& r' r 1 s zArray.__rfloordiv__c C s, | |dd¡}|tkr|S | j |j¡ | S )z5
|
Performs the operation __ilshift__.
|
rw Ú __ilshift__)rD rB r r r r& r& r' r < s
|
zArray.__ilshift__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z5
|
Performs the operation __rlshift__.
|
rw Ú __rlshift__)rD rB rG r r r% r( r` r& r& r' r F s zArray.__rlshift__c C sz | |dd¡}|tkr|S |j}| jdks2|dkr:tdt|dksV|d |d kr^td| j |j¡| jd d
< | S )
|
z5
|
Performs the operation __imatmul__.
|
r^ Ú __imatmul__r& z"@= requires at least one dimensionr éÿÿÿÿéþÿÿÿz-@= cannot change the shape of the input arrayN)rD rB r7 rd rN r rz )r, r: Z other_shaper& r& r' r Q s zArray.__imatmul__c C s4 | |dd¡}|tkr|S | j |j¡}| j |¡S )z5
|
Performs the operation __rmatmul__.
|
r^ Ú __rmatmul__)rD rB r r r% r( r` r& r& r' r g s
|
zArray.__rmatmul__c C s, | |dd¡}|tkr|S | j |j¡ | S )z2
|
Performs the operation __imod__.
|
r^ Ú__imod__)rD rB r r r r& r& r' r s s
|
zArray.__imod__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z2
|
Performs the operation __rmod__.
|
r^ Ú__rmod__)rD rB rG r r r% r( r` r& r& r' r } s zArray.__rmod__c C s, | |dd¡}|tkr|S | j |j¡ | S )z2
|
Performs the operation __imul__.
|
r^ Ú__imul__)rD rB r r r r& r& r' r s
|
zArray.__imul__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z2
|
Performs the operation __rmul__.
|
r^ Ú__rmul__)rD rB rG r r r% r( r` r& r& r' r s zArray.__rmul__c C s, | |dd¡}|tkr|S | j |j¡ | S )z1
|
Performs the operation __ior__.
|
ra Ú__ior__)rD rB r r r r& r& r' r s
|
z Array.__ior__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z1
|
Performs the operation __ror__.
|
ra Ú__ror__)rD rB rG r r r% r( r` r& r& r' r § s z Array.__ror__c C s, | |dd¡}|tkr|S | j |j¡ | S )z2
|
Performs the operation __ipow__.
|
r^ Ú__ipow__)rD rB r r r r& r& r' r ² s
|
zArray.__ipow__c C s0 ddl m} | |dd¡}|tkr&|S ||| S )z2
|
Performs the operation __rpow__.
|
r r r^ Ú__rpow__r r
r& r& r' r¡ ¼ s
|
zArray.__rpow__c C s, | |dd¡}|tkr|S | j |j¡ | S )z5
|
Performs the operation __irshift__.
|
rw Ú __irshift__)rD rB r r¢ r r& r& r' r¢ É s
|
zArray.__irshift__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z5
|
Performs the operation __rrshift__.
|
rw Ú __rrshift__)rD rB rG r r£ r% r( r` r& r& r' r£ Ó s zArray.__rrshift__c C s, | |dd¡}|tkr|S | j |j¡ | S )z2
|
Performs the operation __isub__.
|
r^ Ú__isub__)rD rB r r¤ r r& r& r' r¤ Þ s
|
zArray.__isub__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z2
|
Performs the operation __rsub__.
|
r^ Ú__rsub__)rD rB rG r r¥ r% r( r` r& r& r' r¥ è s zArray.__rsub__c C s, | |dd¡}|tkr|S | j |j¡ | S )z6
|
Performs the operation __itruediv__.
|
r Ú __itruediv__)rD rB r r¦ r r& r& r' r¦ ó s
|
zArray.__itruediv__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z6
|
Performs the operation __rtruediv__.
|
r Ú __rtruediv__)rD rB rG r r§ r% r( r` r& r& r' r§ ý s zArray.__rtruediv__c C s, | |dd¡}|tkr|S | j |j¡ | S )z2
|
Performs the operation __ixor__.
|
ra Ú__ixor__)rD rB r r¨ r r& r& r' r¨ s
|
zArray.__ixor__c C sD | |dd¡}|tkr|S | | |¡\} }| j |j¡}| j |¡S )z2
|
Performs the operation __rxor__.
|
ra Ú__rxor__)rD rB rG r r© r% r( r` r& r& r' r© s zArray.__rxor__r )rh r, Údevicer- c C s. |d k rt d|dkr | S t d|d S )Nz3The stream argument to to_device() is not supportedÚcpuzUnsupported device )rd )r, rª rh r& r& r' Ú to_device s
|
zArray.to_devicer c C s | j jS )z
|
Array API compatible wrapper for :py:meth:`np.ndarray.dtype <numpy.ndarray.dtype>`.
|
|
See its docstring for more information.
|
)r r r1 r& r& r' r $ s z Array.dtypec C s dS )Nr« r& r1 r& r& r' rª - s z Array.devicec C s ddl m} || S )Nr )Úmatrix_transpose)Zlinalgr )r, r r& r& r' ÚmT2 s zArray.mTc C s | j jS )z
|
Array API compatible wrapper for :py:meth:`np.ndarray.ndim <numpy.ndarray.ndim>`.
|
|
See its docstring for more information.
|
)r rE r1 r& r& r' rE 7 s z
|
Array.ndimzTuple[int, ...]c C s | j jS )z
|
Array API compatible wrapper for :py:meth:`np.ndarray.shape <numpy.ndarray.shape>`.
|
|
See its docstring for more information.
|
)r r7 r1 r& r& r' r7 @ s z Array.shapec C s | j jS )z
|
Array API compatible wrapper for :py:meth:`np.ndarray.size <numpy.ndarray.size>`.
|
|
See its docstring for more information.
|
)r Úsizer1 r& r& r' r¯ I s z
|
Array.sizec C s" | j dkrtd| j | jj¡S )z
|
Array API compatible wrapper for :py:meth:`np.ndarray.T <numpy.ndarray.T>`.
|
|
See its docstring for more information.
|
é zwx.T requires x to have 2 dimensions. Use x.mT to transpose stacks of matrices and permute_dims() to permute dimensions.)rE rd r% r( r ÚTr1 r& r& r' r± R s
|
|
zArray.T)N)N)TÚ__name__Ú
|
__module__Ú __qualname__Ú__doc__Ú__annotations__Ú classmethodr( r r/ r8 r9 rD rA Ú staticmethodrG rZ r[ r_ rb re rf rj rk rm rn ro rp rq rr rs rt ru rv rx ry rz r{ r| r} r~ r r r r r r r r r r r r r r r r r r r r r r r r r r¡ r¢ r£ r¤ r¥ r¦ r§ r¨ r© r¬ Úpropertyr rª r® rE r7 r¯ r± Ú __classcell__r& r& r$ r' r , s°
|
*&$ ' ÿ
|
|
|
|
|
|
|
|
|
|
|
|
r )#rµ Ú
|
__future__r rS Úenumr Z_creation_functionsr Z_dtypesr r r r r
|
r r r Útypingr r r r r r ÚtypesÚ_typingr r r Z numpy.typingZnptÚnumpyr r r r& r& r& r' Ú<module> s (
|
