Numpy n-dimensional ops (scatter, gather, one-hot, etc) - View it on GitHub