I think you just need to convert your map variable to a tensor variable, so you can slice it with Aesara variables
import aesara.tensor as at
...
map = at.as_tensor(map)
...
I think you just need to convert your map variable to a tensor variable, so you can slice it with Aesara variables
import aesara.tensor as at
...
map = at.as_tensor(map)
...