numba_inner_parallel

mdcraft.algorithm.accelerated.numba_inner_parallel(qs: ndarray[float], rs: ndarray[float]) ndarray[float][source]

Parallel Numba-accelerated inner product between all possible combinations of multiple one-dimensional NumPy arrays \(\mathbf{q}\) and \(\mathbf{r}\), each with shape \((3,)\).

\[\mathbf{q}_i\cdot\mathbf{r}_j =q_{i1}r_{j1}+q_{i2}r_{j2}+q_{i3}r_{j3}\]
Parameters:
qsnp.ndarray

Multiple vectors \(\mathbf{q}\).

Shape: \((N_q,\,3)\).

rsnp.ndarray

Multiple vectors \(\mathbf{r}\).

Shape: \((N_r,\,3)\).

Returns:
snp.ndarray

Inner products of the vectors, \(\mathbf{q}_i\cdot\mathbf{r}_j\).

Shape: \((N_q,\,N_r)\).