U
|
O ±d¡( ã @ s: d dl mZ d dlmZmZmZmZmZ erTddlm Z m
|
Z
|
m Z m Z m Z d dlmZ ddlmZ d dlZdd Zdddd
|
d d d dddddZdKddddddd d ddddZddddd d dddd Zddddd d dd d d ZdLd ddd d!d"d!d d dd#d$d%Zd&dd'd(d)Zdddddd d dd*d+d,Zdddddd d dd-d.d/Z ddd0d1d!ddd d d2dd3d4d5Z d6d7dd8d9d:d;d<Z ddddd d ddd=d>Z ddddd d dd d?d@Z d dAdd!ddBdCdDZ!d dAdd!ddBdEdFZ"ddddd d dddGdHZ#ddddd d dd dIdJZ$dS )Mé )Ú annotations)Ú TYPE_CHECKINGÚListÚOptionalÚTupleÚUnioné )ÚArrayÚDeviceÚDtypeÚNestedSequenceÚSupportsBufferProtocol)ÚSequence)Ú _all_dtypesNc C s( dt D ]}| |kr d S qtdd S )N)Nz)dtype must be one of the supported dtypes)r Ú
|
ValueError)ÚdtypeÚd© r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/array_api/_creation_functions.pyÚ_check_valid_dtype s r )r ÚdeviceÚcopyzZUnion[Array, bool, int, float, NestedSequence[bool | int | float], SupportsBufferProtocol]zOptional[Dtype]zOptional[Device]z#Optional[Union[bool, np._CopyMode]]r )Úobjr r r Úreturnc C sÐ ddl m} t| |dkr*td||dtjjfkrBtdt| |r|dk rb| j |krbd}|dtjj
|
fkr| tj | j d|d ¡S | S |dkr¸t| tr¸| d
|
ks°| d k r¸td tj| |d }| |¡S )z~
|
Array API compatible wrapper for :py:func:`np.asarray <numpy.asarray>`.
|
|
See its docstring for more information.
|
r ©r ©ÚcpuNúUnsupported device Fz!copy=False is not yet implementedNT)r r l lûÿÿÿ z&Integer out of bounds for array dtypes©r )Ú _array_objectr r r ÚnpZ _CopyModeZ IF_NEEDEDÚNotImplementedErrorÚ
|
isinstancer ZALWAYSÚ_newÚarrayÚ_arrayÚintÚ OverflowErrorÚasarray)r r r r r Úresr r r r( s
|
"r( )r r zOptional[Union[int, float]]zUnion[int, float])ÚstopÚstepÚstartr r r c C sB ddl m} t| |dkr*td|| tj| |||d¡S )z|
|
Array API compatible wrapper for :py:func:`np.arange <numpy.arange>`.
|
|
See its docstring for more information.
|
r r r r )r* r+ r )r r r r r# r Úarange)r, r* r+ r r r r r r r- K s
|
r- zUnion[int, Tuple[int, ...]])Úshaper r r c C s> ddl m} t| |dkr*td|| tj| |d¡S )zz
|
Array API compatible wrapper for :py:func:`np.empty <numpy.empty>`.
|
|
See its docstring for more information.
|
r r r r r )r r r r r# r Úempty©r. r r r r r r r/ a s
|
r/ )Úxr r r c C s@ ddl m} t| |dkr*td|| tj| j|d¡S )z
|
Array API compatible wrapper for :py:func:`np.empty_like <numpy.empty_like>`.
|
|
See its docstring for more information.
|
r r r r r )r r r r r# r Ú
|
empty_liker% ©r1 r r r r r r r2 t s
|
r2 )Úkr r r& z Optional[int])Ún_rowsÚn_colsr4 r r r c C sB ddl m} t| |dkr*td|| tj| |||d¡S )zv
|
Array API compatible wrapper for :py:func:`np.eye <numpy.eye>`.
|
|
See its docstring for more information.
|
r r r r )ÚMr4 r )r r r r r# r Úeye)r5 r6 r4 r r r r r r r8 s
|
r8 Úobject)r1 r c C s ddl m} | t | ¡¡S )Nr r )r r r# r Ú from_dlpack)r1 r r r r r: s r: )r. Ú
|
fill_valuer r r c C sp ddl m} t| |dkr*td|t||rD|jdkrD|j}tj| ||d}|j t
|
krft d| |¡S )zx
|
Array API compatible wrapper for :py:func:`np.full <numpy.full>`.
|
|
See its docstring for more information.
|
r r r r r r zInvalid input to full) r r r r r" Úndimr% r Úfullr r Ú TypeErrorr# )r. r; r r r r) r r r r= s
|
r= )r; r1 r r r c C sX ddl m} t| |dkr*td|tj| j||d}|jtkrNt d|
|
|¡S )z
|
Array API compatible wrapper for :py:func:`np.full_like <numpy.full_like>`.
|
|
See its docstring for more information.
|
r r r r r zInvalid input to full_like) r r r r r Ú full_liker% r r r> r# )r1 r; r r r r) r r r r? » s
|
r? T)r r ÚendpointÚbool)Únumr, r* r r r@ r c C sD ddl m} t| |dkr*td|| tj| ||||d¡S )z
|
Array API compatible wrapper for :py:func:`np.linspace <numpy.linspace>`.
|
|
See its docstring for more information.
|
r r r r )r r@ )r r r r r# r Úlinspace)r, r* rB r r r@ r r r r rC Õ s
|
rC Zxy)ÚindexingÚstrz List[Array])ÚarraysrD r c sR ddl m tdd |D dkr*td fddtjdd |D d | iD S )
|
z
|
Array API compatible wrapper for :py:func:`np.meshgrid <numpy.meshgrid>`.
|
|
See its docstring for more information.
|
r r c S s h | ]
|
}|j qS r r ©Ú.0Úar r r Ú <setcomp>÷ s zmeshgrid.<locals>.<setcomp>z,meshgrid inputs must all have the same dtypec s g | ]} |¡qS r )r# )rH r$ r r r Ú
|
<listcomp>ú s ÿz meshgrid.<locals>.<listcomp>c S s g | ]
|
}|j qS r )r% rG r r r rK ü s rD )r r Úlenr r Úmeshgrid)rD rF r r r rM ì s
|
þrM c C s> ddl m} t| |dkr*td|| tj| |d¡S )zx
|
Array API compatible wrapper for :py:func:`np.ones <numpy.ones>`.
|
|
See its docstring for more information.
|
r r r r r )r r r r r# r Úonesr0 r r r rN s
|
rN c C s@ ddl m} t| |dkr*td|| tj| j|d¡S )z
|
Array API compatible wrapper for :py:func:`np.ones_like <numpy.ones_like>`.
|
|
See its docstring for more information.
|
r r r r r )r r r r r# r Ú ones_liker% r3 r r r rO s
|
rO ©r4 )r1 r4 r c C s4 ddl m} | jdk r td| tj| j|d¡S )zx
|
Array API compatible wrapper for :py:func:`np.tril <numpy.tril>`.
|
|
See its docstring for more information.
|
r r é z)x must be at least 2-dimensional for trilrP )r r r< r r# r Útrilr% ©r1 r4 r r r r rR # s
|
rR c C s4 ddl m} | jdk r td| tj| j|d¡S )zx
|
Array API compatible wrapper for :py:func:`np.triu <numpy.triu>`.
|
|
See its docstring for more information.
|
r r rQ z)x must be at least 2-dimensional for triurP )r r r< r r# r Útriur% rS r r r rT 1 s
|
rT c C s> ddl m} t| |dkr*td|| tj| |d¡S )zz
|
Array API compatible wrapper for :py:func:`np.zeros <numpy.zeros>`.
|
|
See its docstring for more information.
|
r r r r r )r r r r r# r Úzerosr0 r r r rU ? s
|
rU c C s@ ddl m} t| |dkr*td|| tj| j|d¡S )z
|
Array API compatible wrapper for :py:func:`np.zeros_like <numpy.zeros_like>`.
|
|
See its docstring for more information.
|
r r r r r )r r r r r# r Ú
|
zeros_liker% r3 r r r rV R s
|
rV )Nr )N)%Ú
|
__future__r Útypingr r r r r Ú_typingr r
|
r r r Úcollections.abcr Z_dtypesr Únumpyr r r( r- r/ r2 r8 r: r= r? rC rM rN rO rR rT rU rV r r r r Ú<module> sn ó/ üù ü ÿ þù
|
û ú ø ü ÿü ÿ
|
