this is a little tricky? what exactly is the mapping here to?

is `(i,j)` in 2D counts matrix to the index in `self.partial_counts`?

is this sort of like a DOK sparse matrix then?