71 lines
1.9 KiB
Python
71 lines
1.9 KiB
Python
"""
|
|
This type stub file was generated by pyright.
|
|
"""
|
|
|
|
from ._array_object import Array
|
|
from typing import List, Optional, Tuple, Union
|
|
|
|
def concat(arrays: Union[Tuple[Array, ...], List[Array]], /, *, axis: Optional[int] = ...) -> Array:
|
|
"""
|
|
Array API compatible wrapper for :py:func:`np.concatenate <numpy.concatenate>`.
|
|
|
|
See its docstring for more information.
|
|
"""
|
|
...
|
|
|
|
def expand_dims(x: Array, /, *, axis: int) -> Array:
|
|
"""
|
|
Array API compatible wrapper for :py:func:`np.expand_dims <numpy.expand_dims>`.
|
|
|
|
See its docstring for more information.
|
|
"""
|
|
...
|
|
|
|
def flip(x: Array, /, *, axis: Optional[Union[int, Tuple[int, ...]]] = ...) -> Array:
|
|
"""
|
|
Array API compatible wrapper for :py:func:`np.flip <numpy.flip>`.
|
|
|
|
See its docstring for more information.
|
|
"""
|
|
...
|
|
|
|
def permute_dims(x: Array, /, axes: Tuple[int, ...]) -> Array:
|
|
"""
|
|
Array API compatible wrapper for :py:func:`np.transpose <numpy.transpose>`.
|
|
|
|
See its docstring for more information.
|
|
"""
|
|
...
|
|
|
|
def reshape(x: Array, /, shape: Tuple[int, ...], *, copy: Optional[Bool] = ...) -> Array:
|
|
"""
|
|
Array API compatible wrapper for :py:func:`np.reshape <numpy.reshape>`.
|
|
|
|
See its docstring for more information.
|
|
"""
|
|
...
|
|
|
|
def roll(x: Array, /, shift: Union[int, Tuple[int, ...]], *, axis: Optional[Union[int, Tuple[int, ...]]] = ...) -> Array:
|
|
"""
|
|
Array API compatible wrapper for :py:func:`np.roll <numpy.roll>`.
|
|
|
|
See its docstring for more information.
|
|
"""
|
|
...
|
|
|
|
def squeeze(x: Array, /, axis: Union[int, Tuple[int, ...]]) -> Array:
|
|
"""
|
|
Array API compatible wrapper for :py:func:`np.squeeze <numpy.squeeze>`.
|
|
|
|
See its docstring for more information.
|
|
"""
|
|
...
|
|
|
|
def stack(arrays: Union[Tuple[Array, ...], List[Array]], /, *, axis: int = ...) -> Array:
|
|
"""
|
|
Array API compatible wrapper for :py:func:`np.stack <numpy.stack>`.
|
|
|
|
See its docstring for more information.
|
|
"""
|
|
...
|
|
|