jax.lax.add

jax.lax.add(x, y)

逐元素加法: \(x + y\).

此函数直接降级为 stablehlo.add 操作。

参数:

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

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

返回:

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

返回类型:

数组

另请参阅

• jax.numpy.add(): 支持混合 dtype 和秩的 NumPy 样式加法。