jax.lax.mul#

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

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

This function lowers directly to thestablehlo.multiply operation.

Parameters:

• x (ArrayLike) – Input arrays. Must have matching numerical dtypes. If neitheris a scalar,x andy must have the same number of dimensionsand be broadcast compatible.

• y (ArrayLike) – Input arrays. Must have matching numerical dtypes. If neitheris a scalar,x andy must have the same number of dimensionsand be broadcast compatible.

Returns:

An array of the same dtype asx andy containing the productof each pair of broadcasted entries.

Return type:

Array

See also

• jax.numpy.multiply(): NumPy-style multiplication supportinginputs with mixed dtypes and ranks.