Home - News ( United States | United Kingdom | Italy | Germany ) - Football scores

Showing content from https://docs.jax.dev/en/latest/_autosummary/jax.numpy.maximum.html below:

jax.numpy.maximum — JAX documentation

jax.numpy.maximum = <jnp.ufunc 'maximum'>#

Return element-wise maximum of the input arrays.

JAX implementation of numpy.maximum.

Parameters:

• x – input array or scalar.

• y – input array or scalar. Both x and y should either have same shape or be broadcast compatible.

• args (ArrayLike)

• out (None)

• where (None)

Returns:

An array containing the element-wise maximum of x and y.

Return type:

Any

Note

For each pair of elements, jnp.maximum returns:

• larger of the two if both elements are finite numbers.

• nan if one element is nan.

Examples

Inputs with x.shape == y.shape:

>>> x = jnp.array([1, -5, 3, 2])
>>> y = jnp.array([-2, 4, 7, -6])
>>> jnp.maximum(x, y)
Array([1, 4, 7, 2], dtype=int32)

Inputs with broadcast compatibility:

>>> x1 = jnp.array([[-2, 5, 7, 4],
...                 [1, -6, 3, 8]])
>>> y1 = jnp.array([-5, 3, 6, 9])
>>> jnp.maximum(x1, y1)
Array([[-2,  5,  7,  9],
       [ 1,  3,  6,  9]], dtype=int32)

Inputs having nan:

>>> nan = jnp.nan
>>> x2 = jnp.array([nan, -3, 9])
>>> y2 = jnp.array([[4, -2, nan],
...                 [-3, -5, 10]])
>>> jnp.maximum(x2, y2)
Array([[nan, -2., nan],
      [nan, -3., 10.]], dtype=float32)

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