Revert "proper flatten in acc (apache#9619)"

This reverts commit ed823b2.
szha · Feb 7, 2018 · 35d7654 · 35d7654
1 parent 5d4f0d2
commit 35d7654
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/mxnet/metric.py b/python/mxnet/metric.py
@@ -399,7 +399,7 @@ def update(self, labels, preds):
             if pred_label.context != label.context:
                 pred_label = pred_label.as_in_context(label.context)
 
-            self.sum_metric += (pred_label.reshape((-1,)) == label.reshape((-1,))).sum().asscalar()
+            self.sum_metric += (pred_label.flatten() == label.flatten()).sum().asscalar()
             self.num_inst += numpy.prod(pred_label.shape)