doc add relu (#20193)

apache · Apr 20, 2021 · 7dba11a · 7dba11a
1 parent 5da68f7
commit 7dba11a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/docs/python_docs/python/api/npx/index.rst b/docs/python_docs/python/api/npx/index.rst
@@ -82,6 +82,7 @@ More operators
    :toctree: generated/
 
    sigmoid
+   relu
    smooth_l1
    softmax
    log_softmax

diff --git a/src/operator/numpy/np_elemwise_unary_op_basic.cc b/src/operator/numpy/np_elemwise_unary_op_basic.cc
@@ -30,8 +30,11 @@ namespace op {
 
 MXNET_OPERATOR_REGISTER_UNARY(_npx_relu)
 .describe(R"code(Computes rectified linear activation.
+
 .. math::
+
    max(features, 0)
+
 )code" ADD_FILELINE)
 .set_attr<FCompute>("FCompute<cpu>", UnaryOp::Compute<cpu, mshadow_op::relu>)
 .set_attr<nnvm::FGradient>("FGradient", ElemwiseGradUseOut{"_backward_relu"});