jax.lax.rem

jax.lax.rem(x, y)

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

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

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

参数:

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

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

返回值:

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

返回类型:

Array

参见

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