fix test case

PaddlePaddle · Dec 6, 2022 · ad6f08d · ad6f08d
1 parent fbaafab
commit ad6f08d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/paddle/tensor/creation.py b/python/paddle/tensor/creation.py
@@ -1486,7 +1486,7 @@ def diagflat(x, offset=0, name=None):
             return _C_ops.diag(y, offset, padding_value)
 
     if _in_legacy_dygraph():
-        if len(x.shape) <= 1:
+        if len(x.shape) == 1:
             return _legacy_C_ops.diag_v2(
                 x, "offset", offset, "padding_value", padding_value
             )