Fix occasional test failure of TestWalkerAlias

chainer · Jan 23, 2019 · 2d75429 · 2d75429
1 parent 3b014b0
commit 2d75429
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/chainer/utils/walker_alias.py b/chainer/utils/walker_alias.py
@@ -104,12 +104,11 @@ def sample(self, shape):
 
     def sample_xp(self, xp, shape):
         thr_dtype = self.threshold.dtype
-        ps = xp.random.uniform(0, 1, shape).astype(thr_dtype)
-        pb = ps * len(self.threshold)
+        pb = xp.random.uniform(0, len(self.threshold), shape)
         index = pb.astype(numpy.int32)
         left_right = (
             self.threshold[index]
-            < (pb - index.astype(thr_dtype)))
+            < (pb.astype(thr_dtype) - index.astype(thr_dtype)))
         left_right = left_right.astype(numpy.int32)
         return self.values[index * 2 + left_right]