From 1021089cda555d8c0dc348a767aba59d10ab3e62 Mon Sep 17 00:00:00 2001 From: weixing02 Date: Fri, 15 Jun 2018 07:03:44 +0000 Subject: [PATCH] fix --- paddle/fluid/operators/hierarchical_sigmoid_op.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/operators/hierarchical_sigmoid_op.cc b/paddle/fluid/operators/hierarchical_sigmoid_op.cc index 119c437f90d..147374bc546 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]"); -- GitLab