Merge pull request #8310 from guoshengCS/fix-fc-bias

Fix the bias of fc when num_flatten_dims is not 1 in fluid layers
7 years ago · 222155ccbf
parent 1f26dce6e5 a063fc28b6
commit 222155ccbf
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/v2/fluid/layers/nn.py
+++ b/python/paddle/v2/fluid/layers/nn.py
@ -185,7 +185,7 @@ def fc(input,
        helper.append_op(
            type="sum", inputs={"X": mul_results}, outputs={"Out": pre_bias})
    # add bias
-    pre_activation = helper.append_bias_op(pre_bias)
+    pre_activation = helper.append_bias_op(pre_bias, dim_start=num_flatten_dims)
    # add activation
    return helper.append_activation(pre_activation)