Fix det

chainer · Aug 20, 2017 · 271e5fb · 271e5fb
1 parent f519d6c
commit 271e5fb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/chainer/functions/math/det.py b/chainer/functions/math/det.py
@@ -108,6 +108,6 @@ def det(a):
 
     """
     shape = (1, a.shape[0], a.shape[1])
-    batched_a = reshape.Reshape(shape)(a)
+    batched_a = reshape.reshape(a, shape)
     batched_det = BatchDet()(batched_a)
-    return reshape.Reshape(())(batched_det)
+    return reshape.reshape(batched_det, ())