jax.lax.sub

jax.lax.sub(x, y)

逐元素减法：\(x - y\)。

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

参数:

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

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

返回:

一个与 x 和 y 具有相同 dtype 的数组，其中包含每对广播条目的差值。

返回类型:

Array

另请参阅

• jax.numpy.subtract()：NumPy 风格的减法，支持具有混合 dtype 和秩的输入。