jax.lax.cummax

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

计算沿 axis 的累积最大值。

参数:

• operand (Array)

• axis (int)

• reverse (bool)

返回类型:

Array