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

2026-07-03 01:09:13 +08:00 · 2017-12-18 17:41:44 -08:00
parent 47b1f02d3e
commit 24b93b1123
1 changed files with 1 additions and 1 deletions
@@ -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