Array manipulation functions

The following functions conform to Python Array API standard:


Broadcasts one or more dpctl.tensor.usm_ndarrays against one another.

broadcast_to(x, shape)

Broadcast an array to a new shape; returns the broadcasted dpctl.tensor.usm_ndarray as a view.

concat(arrays, axis)

Joins a sequence of arrays along an existing axis.

expand_dims(x, axis)

Expands the shape of an array by inserting a new axis (dimension) of size one at the position specified by axis.

flip(x, axis)

Reverses the order of elements in an array x along the given axis.

moveaxis(x, source, destination)

Moves axes of an array to new positions.

permute_dims(x, axes)

Permute the axes (dimensions) of an array; returns the permuted array as a view.

repeat(x, repeats[, axis])

Repeat elements of an array on a per-element basis.

reshape(x, shape[, order])

Reshapes array x into new shape.

roll(x, shift, axis)

Rolls array elements along a specified axis.

squeeze(x, axis)

Removes singleton dimensions (axes) from array x.

stack(arrays, axis)

Joins a sequence of arrays along a new axis.

tile(x, repetitions)

Repeat an input array x along each axis a number of times given by repetitions.

unstack(x[, axis])

Splits an array in a sequence of arrays along the given axis.