jax.Array.choose

jax.Array.choose#

abstract Array.choose(choices, out=None, mode='raise')[source]#

从多个数组的元素中构造一个数组。

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

参数:
  • self (Array)

  • choices (Sequence[ArrayLike])

  • out (None)

  • mode (str)

返回类型:

Array