next_double_normal_sample Class — pytorch Architecture

Source Code

aten/src/ATen/core/DistributionsHelper.h lines 120–130

template <typename RNG,
          typename = decltype(&RNG::next_double_normal_sample),
          typename = decltype(&RNG::set_next_double_normal_sample)>
C10_HOST_DEVICE bool maybe_get_next_normal_sample(RNG* generator, double* ret) {
  const auto sample = generator->next_double_normal_sample();
  if (!sample.has_value())
    return false;
  *ret = sample.value();
  generator->set_next_double_normal_sample(std::nullopt);
  return true;
}