diff --git a/paddle/fluid/operators/hierarchical_sigmoid_op.cc b/paddle/fluid/operators/hierarchical_sigmoid_op.cc index 119c437f90..147374bc54 100644 --- a/paddle/fluid/operators/hierarchical_sigmoid_op.cc +++ b/paddle/fluid/operators/hierarchical_sigmoid_op.cc @@ -95,7 +95,7 @@ class HierarchicalSigmoidOpMaker : public framework::OpProtoAndCheckerMaker { "[num_classes - 1, D]"); AddInput("Label", "(Tensor, required), The labels of training data. It's a" - "1-D tensor, which the shape is [1, N]"); + "1-D tensor, which the shape is [N, 1]"); AddInput("Bias", "(Tensor, optional), The bias is a tensor with shape" "[1, num_classes - 1]");