jax.lax.mul#

jax.lax.mul(x, y)[source]#

Elementwise multiplication: \(x \times y\).

This function lowers directly to the stablehlo.multiply operation.

Parameters:

  • x (ArrayLike) – Input arrays. Must have matching numerical dtypes. If neither is a scalar, x and y 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 and y must have the same number of dimensions and be broadcast compatible.

Returns:

An array of the same dtype as x and y containing the product of each pair of broadcasted entries.

Return type:

Array

See also