Add tanh to two way ops

elftausend · Nov 19, 2023 · 2c12174 · 2c12174
1 parent 6d9a2d5
commit 2c12174
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/src/two_way_ops/ops/unary.rs b/src/two_way_ops/ops/unary.rs
@@ -91,6 +91,27 @@ impl<C: ToCLSource> ToCLSource for Tan<C> {
     }
 }
 
+pub struct Tanh<C> {
+    pub comb: C,
+}
+
+impl<C> Combiner for Tanh<C> {}
+
+impl<T: Float, C: Eval<T>> Eval<T> for Tanh<C> {
+    #[inline]
+    fn eval(self) -> T {
+        Float::tanh(&self.comb.eval())
+    }
+}
+
+#[cfg(not(feature = "no-std"))]
+impl<C: ToCLSource> ToCLSource for Tanh<C> {
+    #[inline]
+    fn to_cl_source(&self) -> String {
+        format!("tanh({})", self.comb.to_cl_source())
+    }
+}
+
 pub struct Neg<C> {
     pub comb: C,
 }