提交 1021089c 编写于 作者: W weixing02

fix

上级 ee13b396
...@@ -95,7 +95,7 @@ class HierarchicalSigmoidOpMaker : public framework::OpProtoAndCheckerMaker { ...@@ -95,7 +95,7 @@ class HierarchicalSigmoidOpMaker : public framework::OpProtoAndCheckerMaker {
"[num_classes - 1, D]"); "[num_classes - 1, D]");
AddInput("Label", AddInput("Label",
"(Tensor, required), The labels of training data. It's a" "(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", AddInput("Bias",
"(Tensor, optional), The bias is a tensor with shape" "(Tensor, optional), The bias is a tensor with shape"
"[1, num_classes - 1]"); "[1, num_classes - 1]");
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册