added method for getting a set of random integers (dmlc#297)

mshadow/random.h

@@ -81,6 +81,13 @@ class Random<cpu, DType> {
     return rnd_engine_();
   }
 
+  /*!
+   * \brief get a set of random integers
+   */
+  inline void GetRandInt(const Tensor<cpu, 1, unsigned>& dst) {
+    std::generate_n(dst.dptr_, dst.size(0), rnd_engine_);
+  }
+
   /*!
    * \brief generate data from a distribution
    * \param dst destination
@@ -395,6 +402,13 @@ class Random<gpu, DType> {
     status = curandSetPseudoRandomGeneratorSeed(gen_, seed);
     CHECK_EQ(status, CURAND_STATUS_SUCCESS) << "Set CURAND seed failed.";
   }
+  /*!
+   * \brief get a set of random integers
+   */
+  inline void GetRandInt(const Tensor<gpu, 1, unsigned>& dst) {
+    curandStatus_t status = curandGenerate(gen_, dst.dptr_, dst.size(0));
+    CHECK_EQ(status, CURAND_STATUS_SUCCESS) << "CURAND Gen rand ints failed.";
+  }
   /*!
    * \brief generate data from uniform [a,b)
    * \param dst destination
@@ -476,7 +490,7 @@ class Random<gpu, DType> {
     CHECK_EQ(status, CURAND_STATUS_SUCCESS) << "CURAND Gen Uniform double failed."
                                             << " size = " << size;
   }
-  /*! \brief random numbeer generator */
+  /*! \brief random number generator */
   curandGenerator_t gen_;
   /*! \brief templ buffer */
   TensorContainer<gpu, 1, DType> buffer_;