jax.lax.floor

jax.lax.floor(x)

逐元素向下取整：\(\left\lfloor x \right\rfloor\)。

此函数直接降级到stablehlo.floor操作。

参数:

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

返回:

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

返回类型:

数组

另请参阅

• jax.lax.ceil()：四舍五入到正无穷大方向的下一个整数

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