jax.numpy.matvec

jax.numpy.matvec(x1, x2, /)

批量矩阵-向量积。

numpy.matvec() 的 JAX 实现。

参数:

• x1 (Array | ndarray | bool | number | bool | int | float | complex) – 形状为 (..., M, N) 的数组

• x2 (Array | ndarray | bool | number | bool | int | float | complex) – 形状为 (..., N) 的数组。前导维度必须与 x1 的前导维度广播兼容。

返回:

形状为 (..., M) 的数组，包含批量矩阵-向量积。

返回类型:

Array

参见

• jax.numpy.linalg.vecdot()：批量向量积。

• jax.numpy.vecmat()：向量-矩阵积。

• jax.numpy.matmul()：通用矩阵乘法。

示例

简单矩阵-向量积

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

批量矩阵-向量积

>>> x2 = jnp.array([[7, 8, 9],
...                 [5, 6, 7]])
>>> jnp.matvec(x1, x2)
Array([[ 50, 122],
       [ 38,  92]], dtype=int32)