提交 bf7bc127 编写于 作者: T typhoonzero

update

上级 12f0a86f
...@@ -33,7 +33,7 @@ class AucOp : public framework::OperatorWithKernel { ...@@ -33,7 +33,7 @@ class AucOp : public framework::OperatorWithKernel {
PADDLE_ENFORCE_EQ(inference->dims(), label->dims(), PADDLE_ENFORCE_EQ(inference->dims(), label->dims(),
"inference and label should have same shape"); "inference and label should have same shape");
ctx.Output<framework::Tensor>("AUC")->Resize({1}); ctx.Output<framework::LoDTensor>("AUC")->Resize({1});
} }
}; };
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册