jax.lax.floor

jax.lax.floor(x)

Elementwise floor: \(\left\lfloor x \right\rfloor\).

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

参数:

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

返回值:

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

返回类型:

数组

另请参阅

• jax.lax.ceil(): 朝向正无穷方向舍入到下一个整数

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