From 387a5b121583270a153746eb61b64fcbf5ac4801 Mon Sep 17 00:00:00 2001 From: Paul Brossier Date: Thu, 10 Jan 2019 19:12:52 +0100 Subject: [PATCH] [activation] add sigmoid and relu, in-place --- src/ai/activation.c | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 src/ai/activation.c 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] )); + } +} -- 2.11.0