The jacobian will be 2D only if your input and output are 1D.
If you have nD input and mD output, then the Jacobian will be (m+n)D.
If you say that x and h are 2D, what do you mean by  dh_j/dx_i? You need 2 indices to index both h and x here.