From: Paul Brossier <piem@piem.org>
Date: Thu, 10 Jan 2019 18:12:52 +0000 (+0100)
Subject: [activation] add sigmoid and relu, in-place
X-Git-Url: https://git.aubio.org/?a=commitdiff_plain;h=1bba23e40a159d2325cb9c7938093733c12450aa;p=aubio.git

[activation] add sigmoid and relu, in-place
---

diff --git a/src/ai/activation.c b/src/ai/activation.c
new file mode 100644
index 00000000..b27648c4
--- /dev/null
+++ b/src/ai/activation.c
@@ -0,0 +1,21 @@
+#include "aubio_priv.h"
+#include "fmat.h"
+#include "ai/tensor.h"
+
+void aubio_activation_relu(aubio_tensor_t *t)
+{
+  uint_t i;
+  AUBIO_ASSERT(t);
+  for (i = 0; i < t->size; i++) {
+    t->buffer[i] = MAX(0, t->buffer[i]);
+  }
+}
+
+void aubio_activation_sigmoid(aubio_tensor_t *t)
+{
+  uint_t i;
+  AUBIO_ASSERT(t);
+  for (i = 0; i < t->size; i++) {
+    t->buffer[i] = 1. / (1. + EXP( - t->buffer[i] ));
+  }
+}