fix dropout doc

3f5f5ed3 · ceci3 · cd906fc6 · 3f5f5ed3
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/layers/nn.py python/paddle/fluid/layers/nn.py +1 -1

未找到文件。
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@@ -1316,7 +1316,7 @@ def dropout(x,
                                        1. downgrade_in_infer(default), downgrade the outcome at inference

                                           - train: out = input * mask
-                                           - inference: out = input * (1 - dropout_prob)
+                                           - inference: out = input * (1.0 - dropout_prob)

                                           (mask is a tensor same shape with input, value is 0 or 1
                                           ratio of 0 is dropout_prob)