numpy.array_api._manipulation_functions

module documentation
(source)

Undocumented

Function concat Array API compatible wrapper for np.concatenate.
Function expand_dims Array API compatible wrapper for np.expand_dims.
Function flip Array API compatible wrapper for np.flip.
Function permute_dims Array API compatible wrapper for np.transpose.
Function reshape Array API compatible wrapper for np.reshape.
Function roll Array API compatible wrapper for np.roll.
Function squeeze Array API compatible wrapper for np.squeeze.
Function stack Array API compatible wrapper for np.stack.
def concat(arrays: Union[Tuple[Array, ...], List[Array]], /, *, axis: Optional[int] = 0) -> Array: (source)

Array API compatible wrapper for np.concatenate.

See its docstring for more information.

def expand_dims(x: Array, /, *, axis: int) -> Array: (source)

Array API compatible wrapper for np.expand_dims.

See its docstring for more information.

def flip(x: Array, /, *, axis: Optional[Union[int, Tuple[int, ...]]] = None) -> Array: (source)

Array API compatible wrapper for np.flip.

See its docstring for more information.

def permute_dims(x: Array, /, axes: Tuple[int, ...]) -> Array: (source)

Array API compatible wrapper for np.transpose.

See its docstring for more information.

def reshape(x: Array, /, shape: Tuple[int, ...]) -> Array: (source)

Array API compatible wrapper for np.reshape.

See its docstring for more information.

def roll(x: Array, /, shift: Union[int, Tuple[int, ...]], *, axis: Optional[Union[int, Tuple[int, ...]]] = None) -> Array: (source)

Array API compatible wrapper for np.roll.

See its docstring for more information.

def squeeze(x: Array, /, axis: Union[int, Tuple[int, ...]]) -> Array: (source)

Array API compatible wrapper for np.squeeze.

See its docstring for more information.

def stack(arrays: Union[Tuple[Array, ...], List[Array]], /, *, axis: int = 0) -> Array: (source)

Array API compatible wrapper for np.stack.

See its docstring for more information.