From 5f05f4e5c543411c44ea7ccd871779d6307c7ee2 Mon Sep 17 00:00:00 2001
From: liutuo <liutuo@xiaomi.com>
Date: Thu, 7 Jun 2018 19:46:42 +0800
Subject: [PATCH] fix tanh nan bug

---
 mace/kernels/activation.h | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/mace/kernels/activation.h b/mace/kernels/activation.h
index 9979b542..51383ad4 100644
--- a/mace/kernels/activation.h
+++ b/mace/kernels/activation.h
@@ -86,8 +86,7 @@ void DoActivation(const T *input_ptr,
     case TANH:
 #pragma omp parallel for
       for (index_t i = 0; i < size; ++i) {
-        T in_exp = std::exp(-2 * input_ptr[i]);
-        output_ptr[i] = (1 - in_exp) / (1 + in_exp);
+        output_ptr[i] = std::tanh(input_ptr[i]);
       }
       break;
     case SIGMOID:
-- 
GitLab