jax.experimental.pallas.when#

jax.experimental.pallas.when(condition)[源代码]#