jax.lax.slice

jax.lax.slice(operand, start_indices, limit_indices, strides=None)

Wraps XLA’s Slice operator.

Parameters

• operand (Any) –

• start_indices (Sequence[int]) –

• limit_indices (Sequence[int]) –

• strides (Optional[Sequence[int]]) –

Return type

Any