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
Return type

Any