refine paddle.shape (PaddlePaddle#8926)

rsj123 · Apr 15, 2024 · a41023e · a41023e
1 parent 3ee3e4a
commit a41023e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ppdet/modeling/ops.py b/ppdet/modeling/ops.py
@@ -1060,6 +1060,6 @@ def channel_shuffle(x, groups):
 
 
 def get_static_shape(tensor):
-    shape = tensor.shape
+    shape = paddle.shape(tensor)
     shape.stop_gradient = True
     return shape