Changeset 2de864e for src/ai

Timestamp:

Dec 29, 2021, 5:52:00 PM (3 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/crepe

Children:

e75bd80

Parents:

557e0a7

git-author:

Paul Brossier <piem@piem.org> (01/23/19 23:32:36)

git-committer:

Paul Brossier <piem@piem.org> (12/29/21 17:52:00)

Message:

[ai] add some documentation to activation.h

File:

• src/ai/activation.h (modified) (1 diff)

src/ai/activation.h

@@ -23 +23 @@
 #define AUBIO_ACTIVATION_H
 
+/** \file
+
+  Activation functions.
+
+*/
+
 #ifdef __cplusplus
 extern "C" {
 #endif
 
+/** Rectifier linear unit (ReLU)
+
+  \param t input tensor on which to compute the activation function (in-place)
+
+  Upon return, each value \f$x\f$ of the tensor \f$t\f$ will be set to
+  \f$max(0, x)\f$.
+
+*/
 void aubio_activation_relu(aubio_tensor_t *t);
+
+/** compute sigmoid activation
+
+  \param t input tensor on which to compute the activation function (in-place)
+
+  Upon return, each value \f$x\f$ of the tensor \f$t\f$ will be set to
+  \f$\frac{1}{1 + e^{-x}}\f$.
+
+*/
 void aubio_activation_sigmoid(aubio_tensor_t *t);