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

fix GetDims bug
PaddlePaddle · Jan 8, 2018 · 43dab72 · 43dab72
2 parents 6e3cc0c + 8b1a81a
commit 43dab72
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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();