make_mat#
- luas.kronecker_fns.make_mat(r: Array, N_l: int, N_t: int) Array[source]#
Function for converting a vector of shape
(N_l * N_t,)into an array of shape(N_l, N_t).\[r_{i N_l + j} = \mathbf{R}_{ij}\]- Parameters:
r (JAXArray) – Vector of shape
(N_l * N_t,)N_l (int) – Size of wavelength/vertical dimension
N_t (int) – Size of time/horizontal dimension
- Returns:
An array of shape
(N_l, N_t)- Return type:
JAXArray