提交 1f6b9090 编写于 作者: Y Yibing Liu

fix a typo in rank_loss_op

上级 cf4b2db7
...@@ -39,7 +39,7 @@ class RankLossOp : public framework::OperatorWithKernel { ...@@ -39,7 +39,7 @@ class RankLossOp : public framework::OperatorWithKernel {
PADDLE_ENFORCE((label_dims == left_dims) && (left_dims == right_dims), PADDLE_ENFORCE((label_dims == left_dims) && (left_dims == right_dims),
"All inputs must have the same size"); "All inputs must have the same size");
PADDLE_ENFORCE((label_dims.size() == 2) && (label_dims[1] == 1), PADDLE_ENFORCE((label_dims.size() == 2) && (label_dims[1] == 1),
"All inputs must be row vector with size batch_sizex1."); "All inputs must be row vector with size batch_size x 1.");
ctx.Output<framework::LoDTensor>("Out")->Resize(label_dims); ctx.Output<framework::LoDTensor>("Out")->Resize(label_dims);
} }
}; };
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册