module documentation

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, /, *, axis=0):

Array API compatible wrapper for np.concatenate.

See its docstring for more information.

Parameters
arrays:Union[Tuple[Array, ...], List[Array]]Undocumented
axis:Optional[int]Undocumented
Returns
ArrayUndocumented
def expand_dims(x, /, *, axis):

Array API compatible wrapper for np.expand_dims.

See its docstring for more information.

Parameters
x:ArrayUndocumented
axis:intUndocumented
Returns
ArrayUndocumented
def flip(x, /, *, axis=None):

Array API compatible wrapper for np.flip.

See its docstring for more information.

Parameters
x:ArrayUndocumented
axis:Optional[Union[int, Tuple[int, ...]]]Undocumented
Returns
ArrayUndocumented
def permute_dims(x, /, axes):

Array API compatible wrapper for np.transpose.

See its docstring for more information.

Parameters
x:ArrayUndocumented
axes:Tuple[int, ...]Undocumented
Returns
ArrayUndocumented
def reshape(x, /, shape):

Array API compatible wrapper for np.reshape.

See its docstring for more information.

Parameters
x:ArrayUndocumented
shape:Tuple[int, ...]Undocumented
Returns
ArrayUndocumented
def roll(x, /, shift, *, axis=None):

Array API compatible wrapper for np.roll.

See its docstring for more information.

Parameters
x:ArrayUndocumented
shift:Union[int, Tuple[int, ...]]Undocumented
axis:Optional[Union[int, Tuple[int, ...]]]Undocumented
Returns
ArrayUndocumented
def squeeze(x, /, axis):

Array API compatible wrapper for np.squeeze.

See its docstring for more information.

Parameters
x:ArrayUndocumented
axis:Union[int, Tuple[int, ...]]Undocumented
Returns
ArrayUndocumented
def stack(arrays, /, *, axis=0):

Array API compatible wrapper for np.stack.

See its docstring for more information.

Parameters
arrays:Union[Tuple[Array, ...], List[Array]]Undocumented
axis:intUndocumented
Returns
ArrayUndocumented