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

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

jax.numpy.outer — JAX documentation

jax.numpy.outer(a, b, out=None)[source]#

Compute the outer product of two arrays.

JAX implementation of numpy.outer().

Parameters:

• a (Array | ndarray | bool | number | bool | int | float | complex) – first input array, if not 1D it will be flattened.

• b (Array | ndarray | bool | number | bool | int | float | complex) – second input array, if not 1D it will be flattened.

• out (None) – unsupported by JAX.

Returns:

The outer product of the inputs a and b. Returned array will be of shape (a.size, b.size).

Return type:

Array

Examples

>>> a = jnp.array([1, 2, 3])
>>> b = jnp.array([4, 5, 6])
>>> jnp.outer(a, b)
Array([[ 4,  5,  6],
       [ 8, 10, 12],
       [12, 15, 18]], 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