Merge pull request #5334 from kexinzhao/polish_op_doc

Polish activation operator documentation
7 years ago · 39320f00df
parent a8dee3b315 81ba077e7b
commit 39320f00df
2 changed files with 174 additions and 66 deletions
--- a/paddle/operators/activation_op.cc
+++ b/paddle/operators/activation_op.cc
--- a/paddle/operators/activation_op.h
+++ b/paddle/operators/activation_op.h
@ -232,7 +232,7 @@ struct HardShrinkGradFunctor : public BaseActivationFunctor<T> {
  }
 };

-// softshrink(x) = x - lambda, if x > lambda; x + lambda, if x < lambda; 0
+// softshrink(x) = x - lambda, if x > lambda; x + lambda, if x < -lambda; 0
 // otherwise
 template <typename T>
 struct SoftShrinkFunctor : public BaseActivationFunctor<T> {