diff --git a/paddle/fluid/operators/hierarchical_sigmoid_op.cc b/paddle/fluid/operators/hierarchical_sigmoid_op.cc index 119c437f90d4ba3c2801f43eb55a65a52c38b418..147374bc5465dd11871afc93b08ec8a7c6d6b1a8 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]");