*E v d dl mZ d dlmZmZmZmZmZ erddlm Z m
Z
m Z m Z m
Z
d dlmZ ddlmZ d dlZd Zdddd dAdZ dBddddCdZddddDd ZddddEd Z dFd ddd!dGd'ZdHd)ZddddId+ZddddJd,Z ddd-d.dKd2Z d3d4dLd9Z ddddDd:Z ddddEd;Z d d<dMd=Z!d d<dMd>Z"ddddDd?Z#ddddEd@Z$dS )N ) annotations)
TYPE_CHECKINGListOptionalTupleUnion )ArrayDeviceDtypeNestedSequenceSupportsBufferProtocol)Sequence) _all_dtypesNc H dt z D ] }| |u r d S
t d )NNz)dtype must be one of the supported dtypes)r
ValueError)dtypeds w/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/array_api/_creation_functions.py_check_valid_dtyper s<
{
" A:: FF
@
A
AA )r devicecopyobjZUnion[Array, bool, int, float, NestedSequence[bool | int | float], SupportsBufferProtocol]r Optional[Dtype]r Optional[Device]r #Optional[Union[bool, np._CopyMode]]returnr
c ddl m} t | |dvrt d| |dt j j fv rt d t | | rP|
| j |k rd}|dt j j
fv r* |j t j | j
d| S | S |0t | t r| d
k s| d k rt d t j | |
} |j | S )z~
Array API compatible wrapper for :py:func:`np.asarray `.
See its docstring for more information.
r r
cpuNUnsupported device Fz!copy=False is not yet implementedNT)r r l l z&Integer out of bounds for array dtypesr )
_array_objectr
r r np _CopyMode IF_NEEDEDNotImplementedError
isinstancer ALWAYS_newarray_arrayint
OverflowErrorasarray)r r r r r
ress r r3 r3 s, , %$$$$$u
]""9v99::: r|-...!"EFFF#u e!3!3D D",-. . . 5:bhszE J J JKK K
}C--}3==C*DTDT DEEE
*S
&
&
&C 5:c??r )r r stopOptional[Union[int, float]]stepUnion[int, float]startc ddl m} t | |dvrt d| |j t j | ||| S )z|
Array API compatible wrapper for :py:func:`np.arange `.
See its docstring for more information.
r r" r# r% )r5 r7 r )r' r
r r r. r( arange)r9 r5 r7 r r r
s r r; r; K sk %$$$$$u
]""9v99::: 5:biDt5III J JJr shapeUnion[int, Tuple[int, ...]]c ddl m} t | |dvrt d| |j t j | | S )zz
Array API compatible wrapper for :py:func:`np.empty `.
See its docstring for more information.
r r" r# r% r& )r' r
r r r. r( emptyr< r r r
s r r? r? a g %$$$$$u
]""9v99::: 5:bhuE222 3 33r xc ddl m} t | |dvrt d| |j t j | j | S )z
Array API compatible wrapper for :py:func:`np.empty_like `.
See its docstring for more information.
r r" r# r% r& )r' r
r r r. r(
empty_liker0 rB r r r
s r rD rD t i %$$$$$u
]""9v99::: 5:bmAHE::: ; ;;r )kr r n_rowsr1 n_cols
Optional[int]rG c ddl m} t | |dvrt d| |j t j | ||| S )zv
Array API compatible wrapper for :py:func:`np.eye `.
See its docstring for more information.
r r" r# r% )MrG r )r' r
r r r. r( eye)rH rI rG r r r
s r rM rM sk %$$$$$u
]""9v99::: 5:bfVv%@@@ A AAr objectc R ddl m} |j t j | S )Nr r" )r' r
r. r( from_dlpack)rB r
s r rP rP s0 $$$$$$ 5:bnQ'' ( ((r
fill_valuec $ ddl m} t | |dvrt d| t || r|j dk r|j }t j | || }|j t vrt d |j | S )zx
Array API compatible wrapper for :py:func:`np.full `.
See its docstring for more information.
r r" r# r% r r& zInvalid input to full)
r' r
r r r, ndimr0 r( fullr r TypeErrorr. )r< rQ r r r
r4 s r rT rT s %$$$$$u
]""9v99:::*e$$ ' A)=)= &
'%5
1
1
1C
y ## /000 5:c??r c ddl m} t | |dvrt d| t j | j || }|j t vrt d |j
| S )z
Array API compatible wrapper for :py:func:`np.full_like `.
See its docstring for more information.
r r" r# r% r& zInvalid input to full_like) r' r
r r r( full_liker0 r r rU r. )rB rQ r r r
r4 s r rW rW s %$$$$$u
]""9v99:::
,qx5
9
9
9C
y ## 4555 5:c??r T)r r endpointnumrX boolc ddl m} t | |dvrt d| |j t j | |||| S )z
Array API compatible wrapper for :py:func:`np.linspace `.
See its docstring for more information.
r r" r# r% )r rX )r' r
r r r. r( linspace)r9 r5 rY r r rX r
s r r\ r\ sm %$$$$$u
]""9v99::: 5:bk%s%(SSS T TTr xy)indexingarraysr^ str List[Array]c ddl m t d |D dk rt d fdt j d |D d| iD S )z
Array API compatible wrapper for :py:func:`np.meshgrid `.
See its docstring for more information.
r r" c h | ] }|j
S r& .0as r zmeshgrid.. s $ $ $AG $ $ $r z,meshgrid inputs must all have the same dtypec 0 g | ]} j | S rd )r. )rf r/ r
s r
z meshgrid.. s5
5 r c g | ] }|j
S rd )r0 re s r rj z meshgrid.. s "<"<"<18"<"<"`.
See its docstring for more information.
r r" r# r% r& )r' r
r r r. r( onesr@ s r ro ro sg %$$$$$u
]""9v99::: 5:bge5111 2 22r c ddl m} t | |dvrt d| |j t j | j | S )z
Array API compatible wrapper for :py:func:`np.ones_like `.
See its docstring for more information.
r r" r# r% r& )r' r
r r r. r( ones_liker0 rE s r rq rq si %$$$$$u
]""9v99::: 5:bl185999 : ::r rG c ddl m} | j dk rt d |j t j | j | S )zx
Array API compatible wrapper for :py:func:`np.tril `.
See its docstring for more information.
r r" z)x must be at least 2-dimensional for trilrr )r' r
rS r r. r( trilr0 rB rG r
s r ru ru # S %$$$$$vzzDEEE 5:bgah!,,, - --r c ddl m} | j dk rt d |j t j | j | S )zx
Array API compatible wrapper for :py:func:`np.triu `.
See its docstring for more information.
r r" rt z)x must be at least 2-dimensional for triurr )r' r
rS r r. r( triur0 rv s r ry ry 1 rw r c ddl m} t | |dvrt d| |j t j | | S )zz
Array API compatible wrapper for :py:func:`np.zeros `.
See its docstring for more information.
r r" r# r% r& )r' r
r r r. r( zerosr@ s r r{ r{ ? rA r c ddl m} t | |dvrt d| |j t j | j | S )z
Array API compatible wrapper for :py:func:`np.zeros_like `.
See its docstring for more information.
r r" r# r% r& )r' r
r r r. r(
zeros_liker0 rE s r r} r} R rF r )
r r r r r r r r r r
)Nr ) r5 r6 r7 r8 r9 r8 r r r r r r
)r< r= r r r r r r
)rB r
r r r r r r
r ) rH r1 rI rJ rG r1 r r r r r r
)rB rN r r
)
r< r= rQ r8 r r r r r r
)
rQ r8 rB r
r r r r r r
)rY r1 r9 r8 r5 r8 r r r r rX rZ r r
)r_ r
r^ r` r ra )rB r
rG r1 r r
)%
__future__r typingr r r r r _typingr
r r r
r collections.abcr _dtypesr numpyr( r r3 r; r? rD rM rP rT rW r\ rm ro rq ru ry r{ r} rd r r r sX " " " " " " ? > > > > > > > > > > > > > ) )((((( B B B, " #04) ) ) ) ) )^ )- K " #K K K K K K2 " # 4 4 4 4 4 4( .2d
<
<
<
<
<
<$ !B
! #B B B B B B,) ) ) ) " # @ " #
@ " #U U U U U U. .2 . " # 3 3 3 3 3 3( .2d
;
;
;
;
;
; #$ . . . . . . #$ . . . . . ." " # 4 4 4 4 4 4( .2d
<
<
<
<
<
<
<