kron_prod#
- luas.kronecker_fns.kron_prod(A: Array, B: Array, R: Array) Array [source]#
Computes the matrix vector product of the kronecker product of two matrices
A
andB
times a vectorr
, stored as an(N_l, N_t)
arrayR
.\[[\mathbf{A} \otimes \mathbf{B}] \vec{r} = \mathbf{A} \mathbf{R} \mathbf{B}^T\]- Parameters:
A (JAXArray) – Matrix on the left side of the kronecker product.
B (JAXArray) – Matrix on the right side of the kronecker product.
R (JAXArray) – Vector to right multiply, stored as an
(N_l, N_t)
array.
- Returns:
The result of the multiplication as a JAXArray array of shape
(N_l, N_t)
.- Return type:
JAXArray