numpy version of the make_grid function in torch. Dimension of array: NHWC

given the previous states of the box, move it in 1D along x-axis. state: 0-3: current (x, y, z) coordinate of the ce

given the previous states of the glass, move it in 1D along x-axis. update the states of the 5 boxes that form the box: floor, left/r

normalize to [-0.5, 0.5], as in planet.

Random pick a particle up and the drop it for pick_num times