jax.lax.slice

Warning

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

jax.lax.slice¶

jax.lax.slice(operand, start_indices, limit_indices, strides=None)[source]¶

Wraps XLA’s Slice operator.

Parameters

operand (Any) –
start_indices (Sequence[int]) –
limit_indices (Sequence[int]) –
strides (Optional[Sequence[int]]) –

Return type

previous

jax.lax.shift_right_logical

next

jax.lax.slice_in_dim