DeleteΒΆ

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

Abstract Signature:

Delete(arr: Tensor, obj: Union[int, slice, Tensor], axis: int = None)

PyTorch

API: β€”
Strategy: Plugin (torch_delete)

JAX (Core)

API: jax.numpy.delete
Strategy: Direct Mapping

NumPy

API: numpy.delete
Strategy: Direct Mapping

TensorFlow

API: tf.experimental.numpy.delete
Strategy: Direct Mapping

Flax NNX

API: jax.numpy.delete
Strategy: Direct Mapping