jax.Array.searchsorted

jax.Array.searchsorted#

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

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

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

参数:
  • self (Array)

  • v (ArrayLike)

  • side (str)

  • sorter (ArrayLike | None)

  • method (str)

返回类型:

Array