code: @Override
      public void shuffle(INDArray array, Random rnd, int... dimension) {
          shuffle(Collections.singletonList(array), rnd, dimension);
      }
docs: In place shuffle of an ndarray
  along a specified set of dimensions
  
  @param array     the ndarray to shuffle
  @param dimension the dimension to do the shuffle
  @return