jax.Array.choose

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

建構一個從多個陣列的元素中選擇的陣列。

請參閱 jax.numpy.choose() 以取得完整文件。

參數:

• self (Array)

• choices (Sequence[ArrayLike])

• out (None)

• mode (str)

返回類型:

Array