jax.lax.ceil

jax.lax.ceil(x)

逐元素向上取整: \(\left\lceil x \right\rceil\)。

此函数直接降低为 stablehlo.ceil 操作。

参数:

x (ArrayLike) – 输入数组。必须具有浮点类型。

返回:

与 x 形状和 dtype 相同的数组，包含四舍五入到正无穷大方向的下一个整数的值。

返回类型:

Array

另请参阅

• jax.lax.floor(): 向负无穷大方向舍入到下一个整数

• jax.lax.round(): 舍入到最接近的整数