From 64f56bec2904b786cc80626efd37b0eed7201252 Mon Sep 17 00:00:00 2001 From: pangyoki Date: Thu, 10 Sep 2020 08:59:34 +0000 Subject: [PATCH] fix windows ci array len error --- paddle/fluid/operators/multinomial_op.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/operators/multinomial_op.h b/paddle/fluid/operators/multinomial_op.h index cb63fc01972..c025fc3d6ff 100644 --- a/paddle/fluid/operators/multinomial_op.h +++ b/paddle/fluid/operators/multinomial_op.h @@ -31,7 +31,7 @@ void MultinomialFunctor(T* out_data, const T* in_data, const int64_t num_samples, const bool replacement, const int64_t num_categories, const int64_t num_distributions) { - int64_t C = num_categories; + int64_t C = num_categories + 1; T cumulative_probs[C]; std::uniform_real_distribution dist(0, 1); -- GitLab