jax.Array.searchsorted

abstract Array.searchsorted(v, side='left', sorter=None, *, method='scan')

Perform a binary search within a sorted array.

Refer to jax.numpy.searchsorted() for full documentation.

Parameters:

• self (Array)

• v (ArrayLike)

• side (str)

• sorter (ArrayLike | None)

• method (str)

Return type:

Array