Fix the type of out_shape to tuple

cupy · Jul 22, 2020 · 3997341 · 3997341
1 parent be996e8
commit 3997341
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cupy/cutensor.py b/cupy/cutensor.py
@@ -527,8 +527,8 @@ def _try_reduction_routine(x, axis, dtype, out, keepdims, op, alpha, beta):
     in_arg = x
 
     reduce_axis, out_axis = _reduction._get_axis(axis, x.ndim)
-    out_shape = _reduction._get_out_shape(
-        x.shape, reduce_axis, out_axis, keepdims)
+    out_shape = tuple(
+        _reduction._get_out_shape(x.shape, reduce_axis, out_axis, keepdims))
     if out is None:
         out = cupy.empty(out_shape, dtype)
     elif out.shape != out_shape: