diff --git a/paddle/fluid/operators/softmax_with_cross_entropy_op.cc b/paddle/fluid/operators/softmax_with_cross_entropy_op.cc
index 7cf3511806a0d27763f41b88df67a0c4812a3e61..2f4702ed71d6464b056b3fb60b0c895f835bfb2a 100644
--- a/paddle/fluid/operators/softmax_with_cross_entropy_op.cc
+++ b/paddle/fluid/operators/softmax_with_cross_entropy_op.cc
@@ -139,7 +139,6 @@ class SoftmaxWithCrossEntropyOp : public framework::OperatorWithKernel {
     auto loss_dims = logits_dims;
     loss_dims[rank - 1] = 1;
     ctx->SetOutputDim("Loss", loss_dims);
-    // ctx->SetOutputDim("Loss", {logits_dims[0], 1});
 
     ctx->ShareLoD("Logits", /*->*/ "Softmax");
     ctx->ShareLoD("Logits", /*->*/ "Loss");