qttools.kernels.inplace.numba

source package qttools.kernels.inplace.numba

Functions

• iadd — Adds array b to array a at indices inds in-place.

• iadd_obc — Adds array b to array a at indices ind in-place with OBC repetitions.

• isub — Subtracts array b from array a at indices inds in-place.

• isub_obc — Subtracts array b from array a at indices ind in-place with OBC repetitions.

source iadd(a: NDArray, b: NDArray, inds: NDArray) → None

Adds array b to array a at indices inds in-place.

Parameters

• a : NDArray — The array to be updated.

• b : NDArray — The array to be added to a.

• inds : NDArray — The indices at which to add b to a.

source iadd_obc(a: NDArray, b: NDArray, inds: NDArray, k: tuple[float, float], transverse_repetition_grid: tuple[int, int])

Adds array b to array a at indices ind in-place with OBC repetitions.

Parameters

• a : NDArray — The array to be updated.

• b : NDArray — The array to be subtracted from a.

• inds : NDArray — The indices at which to subtract b from a.

• k : tuple[float, float] — The transverse wavevector components.

• transverse_repetition_grid : tuple[int, int] — The transverse repetition grid of the contact.

source isub(a: NDArray, b: NDArray, inds: NDArray) → None

Subtracts array b from array a at indices inds in-place.

Parameters

• a : NDArray — The array to be updated.

• b : NDArray — The array to be subtracted from a.

• inds : NDArray — The indices at which to subtract b from a.

source isub_obc(a: NDArray, b: NDArray, inds: NDArray, k: tuple[float, float], transverse_repetition_grid: tuple[int, int])

Subtracts array b from array a at indices ind in-place with OBC repetitions.

Parameters

• a : NDArray — The array to be updated.

• b : NDArray — The array to be subtracted from a.

• inds : NDArray — The indices at which to subtract b from a.

• k : tuple[float, float] — The transverse wavevector components.

• transverse_repetition_grid : tuple[int, int] — The transverse repetition grid of the contact.