dynamic_slice(operand, start_indices, slice_sizes)¶
Wraps XLA’s DynamicSlice operator.
Any) – an array to slice.
int]) – the size of the slice. Must be a sequence of non-negative integers with length equal to ndim(operand). Inside a JIT compiled function, only static values are supported (all JAX arrays inside JIT must have statically known size).
- Return type
An array containing the slice.