Skip to content

qttools.kernels.datastructure.cupy.dsdbsparse#

source module qttools.kernels.datastructure.cupy.dsdbsparse

Functions

  • find_ranks — Finds the ranks of the indices in the offsets.

source find_ranks(nnz_section_offsets: NDArray, inds: NDArray) → NDArray

Finds the ranks of the indices in the offsets.

Parameters

  • nnz_section_offsets : NDArray — The offsets of the non-zero sections.

  • inds : NDArray — The indices to find the ranks for.

Returns

  • ranks : NDArray — The ranks of the indices in the offsets.