code: public int[] getDims() {
          int[] dims = new int[size()];
          for (int i = 0; i < size(); i++) {
              dims[i] = get(i).getNumStates();
          }
          return dims;
      }
docs: Returns an array holding the number of values the corresponding variables can take on