diff --git a/paddle/fluid/operators/unique_op.h b/paddle/fluid/operators/unique_op.h
index 99793ecd244cf2594a2b0b7462a492bc3f4a27af..66b0543771f4d34faf4941b0a3d5d5b3c97aaf7b 100644
--- a/paddle/fluid/operators/unique_op.h
+++ b/paddle/fluid/operators/unique_op.h
@@ -403,7 +403,10 @@ class UniqueKernel : public framework::OpKernel<T> {
     bool return_index = context.Attr<bool>("return_index");
     bool return_inverse = context.Attr<bool>("return_inverse");
     bool return_counts = context.Attr<bool>("return_counts");
-
+    if (x->numel() == 0) {
+      out->mutable_data<T>(context.GetPlace());
+      return;
+    }
     if (axis_vec.empty()) {
       framework::VisitDataTypeTiny(
           data_type,
diff --git a/paddle/fluid/operators/unstack_op.h b/paddle/fluid/operators/unstack_op.h
index 82118b692707fbb1459980b27573e770c978d521..cfd4d6bce83643982977e626fb809ff87384f264 100644
--- a/paddle/fluid/operators/unstack_op.h
+++ b/paddle/fluid/operators/unstack_op.h
@@ -149,7 +149,7 @@ class UnStackKernel : public framework::OpKernel<T> {
       dx_datas[i] = dx[i]->mutable_data<T>(ctx.GetPlace());
     }
     auto dy_data = dy->data<T>();
-
+    if (dy->numel() == 0) return;
     int pre = 1;
     for (int i = 0; i < axis; ++i) pre *= dy->dims()[i];
     int total_num = dy->numel();
diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py
index 515d4a5c0ef7cd1be858fe47f7a5b307567a01b2..75b0392ab6ae47881e6c3ce1f66b9e12b4930787 100755
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@@ -10315,6 +10315,8 @@ def unstack(x, axis=0, num=None):
     if in_dygraph_mode():
         if num == None:
             num = x.shape[axis]
+        if num == 0:
+            return []
         return _C_ops.unstack(x, num, 'axis', int(axis), 'num', num)
 
     helper = LayerHelper('unstack', **locals())