fixes default type for product of empty shape. (ray-project#1341)

rsepassi · Dec 19, 2017 · 24b93b1 · 24b93b1
1 parent 47b1f02
commit 24b93b1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/ray/experimental/tfutils.py b/python/ray/experimental/tfutils.py
@@ -9,7 +9,7 @@ def unflatten(vector, shapes):
     i = 0
     arrays = []
     for shape in shapes:
-        size = np.prod(shape)
+        size = np.prod(shape, dtype=np.int)
         array = vector[i:(i + size)].reshape(shape)
         arrays.append(array)
         i += size