jax.Array.choose

abstract Array.choose(choices, out=None, mode='raise')

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

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

参数：

• self (Array)

• choices (Sequence[ArrayLike])

• out (None)

• mode (str)

返回类型：

Array