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

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

jax.numpy.vecmat — JAX documentation

jax.numpy.vecmat(x1, x2, /)[source]#

Batched conjugate vector-matrix product.

JAX implementation of numpy.vecmat().

Parameters:

• x1 (Array | ndarray | bool | number | bool | int | float | complex) – array of shape (..., M).

• x2 (Array | ndarray | bool | number | bool | int | float | complex) – array of shape (..., M, N). Leading dimensions must be broadcast-compatible with leading dimensions of x1.

Returns:

An array of shape (..., N) containing the batched conjugate vector-matrix product.

Return type:

Array

Examples

Simple vector-matrix product:

>>> x1 = jnp.array([[1, 2, 3]])
>>> x2 = jnp.array([[4, 5],
...                 [6, 7],
...                 [8, 9]])
>>> jnp.vecmat(x1, x2)
Array([[40, 46]], dtype=int32)

Batched vector-matrix product:

>>> x1 = jnp.array([[1, 2, 3],
...                 [4, 5, 6]])
>>> jnp.vecmat(x1, x2)
Array([[ 40,  46],
       [ 94, 109]], 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