Remove checking 'numel' of 'X' and 'Y'

emailweixu-patch-1
wanghaoshuang 7 years ago
parent ef8cb8f624
commit 67e4ff84d1

@ -58,8 +58,8 @@ class CompareOpInferShape : public framework::InferShapeBase {
comment.type);
auto dim_x = context->GetInputDim("X");
auto dim_y = context->GetInputDim("Y");
PADDLE_ENFORCE_EQ(framework::product(dim_x), framework::product(dim_y),
"The number of elements in X and Y should be same");
PADDLE_ENFORCE_GE(dim_x.size(), dim_y.size(),
"The size of dim_y should not be greater than dim_x's.");
context->SetOutputDim("Out", context->GetInputDim("X"));
context->ShareLoD("X", "Out");

Loading…
Cancel
Save