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

fix

上级 ee13b396
......@@ -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]");
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册