Kernels

The file kernels.py contains the loops that are the core of the particles propagation based on their coordinates. Different batches are available, to allow a matrix (order 1) propagation, a tensor (order 2) propagation or a matrix followed by a tensor (orders 1+2) propagations.

Functions

batched_vector_matrix(b1, b2, matrix)	param b1: a numpy array containing all the particles
batched_vector_matrix_tensor(b1, b2, matrix, ...)	param b1:
batched_vector_tensor(b1, b2, tensor)	param b1:
matrix_matrix(m1, m2)
njit(*args, **kws)	Equivalent to jit(nopython=True)