Function randn

utils/array_utils.hpp:213–219

* @brief Populate the array with random floats, generated from a Gaussian distribution. * @param a The array to populate. * @param N The number of elements in the array. * @param gen The random number generator. * @param mean The mean of the Gaussian distribution. * @param std The standard deviation of the Gaussian distribution. */

 * @param std The standard deviation of the Gaussian distribution.
 */
inline void randn(float *a, size_t N, std::mt19937 &gen, float mean = 0.0,
                  float std = 1.0) {
  std::normal_distribution<float> dist(mean, std);
  for (int i = 0; i < N; i++) {
    a[i] = static_cast<float>(dist(gen));
  }
}

inline void randn(half *a, size_t N, std::mt19937 &gen, float mean = 0.0,
                  float std = 1.0) {