multiply_by

multiply_by(x, by, axis=None)[source]

Broadcast multiply array x by array by along a specified axis.

This function tiles by to match the dimensions of x implied by block multiplication.

Parameters:

  • x (jnp.ndarray) – Input array.

  • by (jnp.ndarray) – Array to multiply with.

  • axis (int, optional) – Axis along which to apply the multiplication (0 for rows, 1 for columns).

Returns:

Result of the multiplication.

Return type:

jnp.ndarray

Raises:

ValueError – If the shape of the specified axis is not divisible by the length of by.