jax.Array.searchsorted

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

在一个已排序的数组中执行二分查找。

请参阅 jax.numpy.searchsorted() 获取完整文档。

参数:

• self (Array)

• v (ArrayLike)

• side (str)

• sorter (ArrayLike | None)

• method (str)

返回类型:

Array