Computes the sizes of states and points chunks

Parameters¶

n_states: int

The number of states

n_targets: int

The number of point targets

Returns¶

chunk_sizes_states: numpy.ndarray

The sizes of all states chunks, shape: (n_chunks_states,)

chunk_sizes_targets: numpy.ndarray

The sizes of all targets chunks, shape: (n_chunks_targets,)