jax.lax.floor

jax.lax.floor(x)

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

参数:

x (ArrayLike)

返回类型:

Array