Fix accuracy_op.

8 years ago · d757c1cd02
parent 48000a8f2d
commit d757c1cd02
1 changed files with 1 additions and 1 deletions
--- a/paddle/operators/accuracy_op.cc
+++ b/paddle/operators/accuracy_op.cc
@ -34,7 +34,7 @@ class AccuracyOp : public framework::OperatorWithKernel {
    PADDLE_ENFORCE_EQ(inference->dims()[0], label->dims()[0],
                      "inference size must be the same as label size");

-    ctx.Output<Tensor>("Accuracy")->Resize({1});
+    ctx.Output<framework::LoDTensor>("Accuracy")->Resize({1});
  }
 };