jax.numpy.delete#

jax.numpy.delete(arr, obj, axis=None)[source]#

Return a new array with sub-arrays along an axis deleted. For a one

LAX-backend implementation of numpy.delete().

Original docstring below.

dimensional array, this returns those entries not returned by arr[obj].

Parameters:
  • arr (array_like) – Input array.

  • obj (slice, int or array of ints) –

    Indicate indices of sub-arrays to remove along the specified axis.

    Changed in version 1.19.0: Boolean indices are now treated as a mask of elements to remove, rather than being cast to the integers 0 and 1.

  • axis (int, optional) – The axis along which to delete the subarray defined by obj. If axis is None, obj is applied to the flattened array.

Returns:

out – A copy of arr with the elements specified by obj removed. Note that delete does not occur in-place. If axis is None, out is a flattened array.

Return type:

ndarray