guochaorong-patch-1
weixing02 7 years ago
parent ee13b396f2
commit 1021089cda

@ -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]");

Loading…
Cancel
Save