mirror of https://github.com/tracel-ai/burn.git
add tanh (#733)
This commit is contained in:
parent
7c34e21424
commit
f253f19b4e
|
@ -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> {
|
fn tanh<const D: usize>(tensor: FloatTensor<Self, D>) -> FloatTensor<Self, D> {
|
||||||
// TODO submit an issue at Candle
|
CandleTensor::new(tensor.tensor.tanh().unwrap())
|
||||||
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())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn erf<const D: usize>(tensor: FloatTensor<Self, D>) -> FloatTensor<Self, D> {
|
fn erf<const D: usize>(tensor: FloatTensor<Self, D>) -> FloatTensor<Self, D> {
|
||||||
|
|
Loading…
Reference in New Issue