return lax.cond(i % 2 == 0, arr1, lambda arr1: arr1.at[i].set(arr1[i] + 1), arr1, ... This API quickly gets unreadable with deeper nested loops.
確定! 回上一頁