Get the position of grid nodes

hello yuanming
I want to get the (x,y,z) position of each grid nodes, here is the code I added to ‘grid_op’ function of taichi (as the red box shows)

However i cannot get what i want, I think this is because I don’t know how the grid nodes are arranged in ‘ti.grouped’. can you help me to figure it out, Thank you!

Could you try

grid_pos = self.dx * I # I is simply ti.Vector(3, dt=ti.i32)

?