| 152 | |
| 153 | template <typename U> |
| 154 | void fill_multinomial( |
| 155 | Xoroshiro128plus* rng, U* probs, dt_int32* dst, size_t num_groups, |
| 156 | size_t num_samples, size_t len_probs, bool replacement) { |
| 157 | if (!replacement) { |
| 158 | fill_multinomial_without_replacement( |
| 159 | rng, probs, dst, num_groups, num_samples, len_probs); |
| 160 | return; |
| 161 | } |
| 162 | for (size_t i = 0; i < num_groups; ++i) { |
| 163 | for (size_t j = 0; j < num_samples; ++j) { |
| 164 | U u = uniform_sample<U>(rng); |
| 165 | U cumsum_res = U(0); |
| 166 | for (size_t k = 0; k < len_probs; ++k) { |
| 167 | cumsum_res += probs[i * len_probs + k]; |
| 168 | if (u <= cumsum_res) { |
| 169 | dst[i * num_samples + j] = k; |
| 170 | break; |
| 171 | } |
| 172 | } |
| 173 | } |
| 174 | } |
| 175 | } |
| 176 | |
| 177 | template <typename T, typename U> |
| 178 | void fill_poisson(Xoroshiro128plus* rng, U* dst, U* lam, size_t size) { |
nothing calls this directly
no test coverage detected