pydims.take#

pydims.take(x, /, indices)[source]#

Returns elements of an array along an axis.

The indices must be 1-D and their single dimension defines the axis along which to take elements.

Parameters:

  • x (TypeVar(DimArr, bound= DimensionedArray)) – Input array.

  • indices (DimensionedArray) – Array of indices to extract from the input array. Must be 1-D.

Returns:

TypeVar(DimArr, bound= DimensionedArray) – Array containing the elements of the input array at the specified indices.