jax.lax.rem

jax.lax.rem(x, y)

逐元素余数：\(x \bmod y\)。

此函数直接降低到 stablehlo.remainder 操作。结果的符号取自被除数，结果的绝对值始终小于除数的绝对值。

整数除法溢出（除以零的余数或 INT_SMIN 除以 -1 的余数）会产生一个实现定义的**值**。

参数:

• x (ArrayLike) – 输入数组。必须具有匹配的 int 或 float 数据类型。如果两者都不是标量，则 x 和 y 必须具有相同数量的维度且兼容广播。

• y (ArrayLike) – 输入数组。必须具有匹配的 int 或 float 数据类型。如果两者都不是标量，则 x 和 y 必须具有相同数量的维度且兼容广播。

返回:

一个与 x 和 y 具有相同数据类型的数组，其中包含余数。

返回类型:

Array

另请参阅

• jax.numpy.remainder()：具有不同符号语义的 NumPy 风格余数。