jax.lax.add#
- jax.lax.add(x, y)[source]#
Elementwise addition: \(x + y\).
This function lowers directly to the stablehlo.add operation.
- Parameters:
x (ArrayLike) – Input arrays. Must have matching numerical dtypes. If neither is a scalar,
x
andy
must have the same number of dimensions and be broadcast compatible.y (ArrayLike) – Input arrays. Must have matching numerical dtypes. If neither is a scalar,
x
andy
must have the same number of dimensions and be broadcast compatible.
- Returns:
An array of the same dtype as
x
andy
containing the sum of each pair of broadcasted entries.- Return type:
See also
jax.numpy.add()
: NumPy-style addition supporting inputs with mixed dtypes and ranks.