Function ids_from_pos

examples/graphics/gravity_sim.cpp:68–70

}

af::array ids_from_pos(vector<af::array> &pos) {
    return (pos[0].as(u32) * height) + pos[1].as(u32);
}

af::array ids_from_3D(vector<af::array> &pos, float Rx, float Ry, float Rz) {
    af::array x0 = (pos[0] - width / 2);