diff --git a/mace/kernels/activation.h b/mace/kernels/activation.h index 9979b5422c5386c5b11dac934e55bd785d85fe88..51383ad44fe0c737acc3229287921e7c47b0173e 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: