jax.Array.searchsorted

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

在已排序的数组中执行二进制搜索。

完整文档请参考 jax.numpy.searchsorted()。

参数:

• self (Array)

• v (ArrayLike)

• side (str)

• sorter (ArrayLike | None)

• method (str)

返回类型:

Array