fix prelu weight shape for NHWC of static mode (#38310)

PaddlePaddle · Dec 27, 2021 · 0e23ac5 · 0e23ac5 · paddle-bot-old · Dec 29, 2021
1 parent 6648a86
commit 0e23ac5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py
@@ -9864,7 +9864,7 @@ def prelu(x, mode, param_attr=None, data_format="NCHW", name=None):
         #NOTE(zhiqiu): Revert shape to [1, channel, 1, 1] for compatibility with saved model of old version.
         #NOTE(GuoxiaWang): support NHWC data format
         if data_format == 'NHWC':
-            alpha_shape = [1, 1, 1, x.shape[1]]
+            alpha_shape = [1, 1, 1, x.shape[-1]]
         else:
             alpha_shape = [1, x.shape[1], 1, 1]