jax.lax.mul

Contents

jax.lax.mul#

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

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

Parameters:
  • x (jax.typing.ArrayLike)

  • y (jax.typing.ArrayLike)

Return type:

Array