jax.lax.select#

jax.lax.select(pred, on_true, on_false)[source]#

Wraps XLA’s Select operator.

Parameters
Return type

Any