fix p_norm with empty shape (#29500)

5 years ago · 60bfd308ab
parent b9e926b8e5
commit 60bfd308ab
1 changed files with 3 additions and 0 deletions
--- a/paddle/fluid/operators/p_norm_op.cc
+++ b/paddle/fluid/operators/p_norm_op.cc
@ -116,6 +116,9 @@ class PnormOp : public framework::OperatorWithKernel {
      for (int i = 0; i < x_dim.size(); ++i) {
        if (i != axis) reduce_dims.emplace_back(x_dim[i]);
      }
+      if (reduce_dims.size() == 0) {
+        reduce_dims.emplace_back(1);
+      }
    }
    x_dim[axis] = 1;