module documentation

Undocumented

Function arange Array API compatible wrapper for np.arange.
Function asarray Array API compatible wrapper for np.asarray.
Function empty Array API compatible wrapper for np.empty.
Function empty_like Array API compatible wrapper for np.empty_like.
Function eye Array API compatible wrapper for np.eye.
Function from_dlpack Undocumented
Function full Array API compatible wrapper for np.full.
Function full_like Array API compatible wrapper for np.full_like.
Function linspace Array API compatible wrapper for np.linspace.
Function meshgrid Array API compatible wrapper for np.meshgrid.
Function ones Array API compatible wrapper for np.ones.
Function ones_like Array API compatible wrapper for np.ones_like.
Function tril Array API compatible wrapper for np.tril.
Function triu Array API compatible wrapper for np.triu.
Function zeros Array API compatible wrapper for np.zeros.
Function zeros_like Array API compatible wrapper for np.zeros_like.
Function _check_valid_dtype Undocumented
def arange(start: Union[int, float], /, stop: Optional[Union[int, float]] = None, step: Union[int, float] = 1, *, dtype: Optional[Dtype] = None, device: Optional[Device] = None) -> Array: (source)

Array API compatible wrapper for np.arange.

See its docstring for more information.

def asarray(obj: Union[Array, bool, int, float, NestedSequence[(bool|int)|float], SupportsBufferProtocol], /, *, dtype: Optional[Dtype] = None, device: Optional[Device] = None, copy: Optional[Union[bool, np._CopyMode]] = None) -> Array: (source)

Array API compatible wrapper for np.asarray.

See its docstring for more information.

def empty(shape: Union[int, Tuple[int, ...]], *, dtype: Optional[Dtype] = None, device: Optional[Device] = None) -> Array: (source)

Array API compatible wrapper for np.empty.

See its docstring for more information.

def empty_like(x: Array, /, *, dtype: Optional[Dtype] = None, device: Optional[Device] = None) -> Array: (source)

Array API compatible wrapper for np.empty_like.

See its docstring for more information.

def eye(n_rows: int, n_cols: Optional[int] = None, /, *, k: int = 0, dtype: Optional[Dtype] = None, device: Optional[Device] = None) -> Array: (source)

Array API compatible wrapper for np.eye.

See its docstring for more information.

def from_dlpack(x: object, /) -> Array: (source)

Undocumented

def full(shape: Union[int, Tuple[int, ...]], fill_value: Union[int, float], *, dtype: Optional[Dtype] = None, device: Optional[Device] = None) -> Array: (source)

Array API compatible wrapper for np.full.

See its docstring for more information.

def full_like(x: Array, /, fill_value: Union[int, float], *, dtype: Optional[Dtype] = None, device: Optional[Device] = None) -> Array: (source)

Array API compatible wrapper for np.full_like.

See its docstring for more information.

def linspace(start: Union[int, float], stop: Union[int, float], /, num: int, *, dtype: Optional[Dtype] = None, device: Optional[Device] = None, endpoint: bool = True) -> Array: (source)

Array API compatible wrapper for np.linspace.

See its docstring for more information.

def meshgrid(*arrays: Array, indexing: str = 'xy') -> List[Array]: (source)

Array API compatible wrapper for np.meshgrid.

See its docstring for more information.

def ones(shape: Union[int, Tuple[int, ...]], *, dtype: Optional[Dtype] = None, device: Optional[Device] = None) -> Array: (source)

Array API compatible wrapper for np.ones.

See its docstring for more information.

def ones_like(x: Array, /, *, dtype: Optional[Dtype] = None, device: Optional[Device] = None) -> Array: (source)

Array API compatible wrapper for np.ones_like.

See its docstring for more information.

def tril(x: Array, /, *, k: int = 0) -> Array: (source)

Array API compatible wrapper for np.tril.

See its docstring for more information.

def triu(x: Array, /, *, k: int = 0) -> Array: (source)

Array API compatible wrapper for np.triu.

See its docstring for more information.

def zeros(shape: Union[int, Tuple[int, ...]], *, dtype: Optional[Dtype] = None, device: Optional[Device] = None) -> Array: (source)

Array API compatible wrapper for np.zeros.

See its docstring for more information.

def zeros_like(x: Array, /, *, dtype: Optional[Dtype] = None, device: Optional[Device] = None) -> Array: (source)

Array API compatible wrapper for np.zeros_like.

See its docstring for more information.

def _check_valid_dtype(dtype): (source)

Undocumented