jax.lax.cummax

jax.lax.cummax(operand, axis=0, reverse=False)

計算沿著 axis 的累積最大值。

參數:

• operand (Array)

• axis (int)

• reverse (bool)

回傳類型:

Array