jax.lax.slice_in_dim

Warning

This page was created from a pull request (#9655).

jax.lax.slice_in_dim¶

jax.lax.slice_in_dim(operand, start_index, limit_index, stride=1, axis=0)[source]¶

Convenience wrapper around slice applying to only one dimension.

Parameters

operand (Any) –
start_index (Optional[int]) –
limit_index (Optional[int]) –
stride (int) –
axis (int) –

Return type

previous

jax.lax.slice

next

jax.lax.sign