enhance error info for py_func (#30138)

* enhance error info for py_func * update
4 years ago · 91a8a25721
parent 6aa82e03ac
commit 91a8a25721
1 changed files with 3 additions and 1 deletions
--- a/paddle/fluid/operators/py_func_op.cc
+++ b/paddle/fluid/operators/py_func_op.cc
@ -112,7 +112,9 @@ static void CallPythonFunc(py::object *callable,
      out->ShareDataWith(*py_out_tensor);
    } catch (py::cast_error &) {
      PADDLE_THROW(platform::errors::InvalidArgument(
-          "The %d-th output must be LoDTensor.", i));
+          "py::cast to LoDTensor error. The %d-th output expection is "
+          "LoDTensor",
+          i));
    }
  }
 }