fixed

seiichiinoue · Jun 23, 2020 · b26192e · b26192e
1 parent eea14b5
commit b26192e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/common/functions.py b/common/functions.py
@@ -23,7 +23,7 @@ def softmax(x):
 
 
 def cross_entropy_error(y, t):
-    if t.ndim == 2:
+    if t.ndim == 2 and t[0].size == 1:
         t = t.reshape(t.size)
     if y.ndim == 1:
         t = t.reshape(1, t.size)