jax.lax.mulhi

Contents

jax.lax.mulhi#

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

Elementwise multiply-high: high bits of \(x \times y\).

For N-bit integer inputs, this function computes the upper N bits of the full 2N-bit product.

Parameters:

  • x (ArrayLike) – Input arrays. Must have an integer dtype. 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 an integer dtype. 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 most significant N bits of the 2N-bit product of each pair of broadcasted entries.

Return type:

Array

Contents