unexpected density shape

[oeur]

In the density computation, an unexpected shape for density is returned. I expected the shape of the input position indexed to exclude the 0th, [1:]. Instead, the input shape is (3,60,60) and the outputted shape returns (3600,), when documentation says it should be (60,60). Hope I explained this well! Thanks!!

[adrn]

Thanks! Working on this now