Unexpected behaviour of indexing?

[BurgerAndreas]

Hi,

Thanks again for this library!

1. Am I right in that n_states / indexing can be used to implement the sparse fixed-point correction of DEQ Optical Flow?

2. If yes, I am confused about the output in this example:

from torchdeq import get_deq

# Settings from `DEQ Optical Flow` paper
args = {
    "n_states": 2,
    "f_max_iter": 24,
}

deq = get_deq(args)

print('deq.indexing: ', deq.indexing)

Output: deq.indexing: [12, 12] Expected output: [8, 16] (uniformly sample between 0 and 24)

Am I missinterpreting?