array_slice

array_slice#

Signature#

nabla.array_slice(arg: 'Array', slices: 'list[slice]', squeeze_axes: 'list[int] | None') -> 'Array'

Description#

Slice an array along specified dimensions.

Parameters#

  • arg (Input array to slice): slices: List of slice objects defining the slicing for each dimension squeeze_axes: List of axes that should be squeezed (for JAX compatibility)

Returns#

  • Sliced array:

Contents