Trait coaster_nn::TanhPointwise

pub trait TanhPointwise<F>: NN<F> {
    fn tanh_pointwise(&self, x: &mut SharedTensor<F>) -> Result<(), Error>;
    fn tanh_pointwise_grad(
        &self, 
        x: &SharedTensor<F>, 
        x_diff: &mut SharedTensor<F>
    ) -> Result<(), Error>;
}

Provides the functionality for pointwise ReLU operations (overwrites the input with the result of the operation).

Required Methods

fn tanh_pointwise(&self, x: &mut SharedTensor<F>) -> Result<(), Error>

Computes the hyperbolic Tangent over the input Tensor x.

Saves the result back to x.

fn tanh_pointwise_grad(
    &self,
    x: &SharedTensor<F>,
    x_diff: &mut SharedTensor<F>
) -> Result<(), Error>

Computes the gradient of tanh over the input Tensor x.

Saves the result back to x_diff.

Implementors

• impl TanhPointwise<f32> for Backend<Native>
• impl TanhPointwise<f64> for Backend<Native>
• impl<T> TanhPointwise<T> for Backend<Cuda> where
    T: Float + Default + DataTypeInfo,