From f253f19b4e485ef55e4f158592e7e3a1ebbe85ae Mon Sep 17 00:00:00 2001
From: Louis Fortier-Dubois <louisfd94@gmail.com>
Date: Wed, 30 Aug 2023 10:00:50 -0400
Subject: [PATCH] add tanh (#733)

---
 burn-candle/src/ops/tensor.rs | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/burn-candle/src/ops/tensor.rs b/burn-candle/src/ops/tensor.rs
index 095f94226..565726db5 100644
--- a/burn-candle/src/ops/tensor.rs
+++ b/burn-candle/src/ops/tensor.rs
@@ -389,10 +389,7 @@ impl<F: FloatCandleElement, I: IntCandleElement> TensorOps<CandleBackend<F, I>>
     }
 
     fn tanh<const D: usize>(tensor: FloatTensor<Self, D>) -> FloatTensor<Self, D> {
-        // TODO submit an issue at Candle
-        let e_x = tensor.tensor.exp().unwrap();
-        let e_minus_x = tensor.tensor.neg().unwrap().exp().unwrap();
-        CandleTensor::new(((e_x.clone() - e_minus_x.clone()).unwrap() / (e_x + e_minus_x)).unwrap())
+        CandleTensor::new(tensor.tensor.tanh().unwrap())
     }
 
     fn erf<const D: usize>(tensor: FloatTensor<Self, D>) -> FloatTensor<Self, D> {