Multiply two arrays element-wise.
JAX implementation of numpy.multiply
. This is a universal function, and supports the additional APIs described at jax.numpy.ufunc
. This function provides the implementation of the *
operator for JAX arrays.
x – arrays to multiply. Must be broadcastable to a common shape.
y – arrays to multiply. Must be broadcastable to a common shape.
args (ArrayLike)
out (None)
where (None)
Array containing the result of the element-wise multiplication.
Any
Examples
Calling multiply
explicitly:
>>> x = jnp.arange(4) >>> jnp.multiply(x, 10) Array([ 0, 10, 20, 30], dtype=int32)
Calling multiply
via the *
operator:
>>> x * 10 Array([ 0, 10, 20, 30], dtype=int32)
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.4