jax.lax.div

jax.lax.div(x, y)

元素级除法: \(x \over y\)。

此函数直接降低为 stablehlo.divide 操作。

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

参数:

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

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

返回:

一个与 x 和 y 具有相同数据类型的数组，其中包含每对广播条目的商。对于整数输入，任何小数部分都会被丢弃。

返回类型:

Array

另请参阅

• jax.numpy.divide(): 支持混合数据类型和秩的 NumPy 风格的真除法。

• jax.numpy.floor_divide(): 支持混合数据类型和秩的 NumPy 风格的地板除法。