From 08de0128b667098b06f93f411f139e34dd2dee62 Mon Sep 17 00:00:00 2001 From: Robert David Date: Thu, 20 Feb 2025 13:28:58 -0800 Subject: [PATCH] Fix head comment: was referring to nonexistent parameters. PiperOrigin-RevId: 729231457 --- jax/_src/lax/slicing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jax/_src/lax/slicing.py b/jax/_src/lax/slicing.py index 2d6182eade73..7255e038ad9d 100644 --- a/jax/_src/lax/slicing.py +++ b/jax/_src/lax/slicing.py @@ -939,7 +939,7 @@ def index_in_dim(operand: Array | np.ndarray, index: int, axis: int = 0, keepdims: bool = True) -> Array: """Convenience wrapper around :func:`lax.slice` to perform int indexing. - This is effectively equivalent to ``operand[..., start_index:limit_index:stride]`` + This is effectively equivalent to ``operand[..., index]`` with the indexing applied on the specified axis. Args: