提交 bf7bc127 编写于 作者: T typhoonzero

update

上级 12f0a86f
......@@ -33,7 +33,7 @@ class AucOp : public framework::OperatorWithKernel {
PADDLE_ENFORCE_EQ(inference->dims(), label->dims(),
"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.
先完成此消息的编辑!
想要评论请 注册