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

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

Loading…
Cancel
Save