Merge pull request #7323 from jacquesqiao/fix-GetDims-bug

fix GetDims bug

Merge pull request #7323 from jacquesqiao/fix-GetDims-bug
fix GetDims bug
43dab723 · emailweixu · GitHub · 6e3cc0c9 · 8b1a81a9 · 43dab723
显示空白变更内容
内联并排

Showing with 3 addition and 1 deletion

paddle/framework/operator.cc paddle/framework/operator.cc +3 -1

未找到文件。
--- a/paddle/framework/operator.cc
+++ b/paddle/framework/operator.cc
@@ -74,7 +74,9 @@ void UseALL() {

 static DDim GetDims(const Scope& scope, const std::string& name) {
  Variable* var = scope.FindVar(name);
-  if (var->IsType<LoDTensor>()) {
+  if (var == nullptr) {
+    return DDim({-1});
+  } else if (var->IsType<LoDTensor>()) {
    return var->Get<LoDTensor>().dims();
  } else if (var->IsType<SelectedRows>()) {
    return var->Get<SelectedRows>().GetCompleteDims();