Struct Tanhshrink 

pub struct Tanhshrink<D: Float, T, const RANK: usize> { /* private fields */ }

Tanhshrink activation layer: x - tanh(x).

Implementations

impl<D: Float, T, const RANK: usize> Tanhshrink<D, T, RANK>

pub fn new() -> Self

Creates a new Tanhshrink layer.

Trait Implementations

impl<D: Float, T, const RANK: usize> Default for Tanhshrink<D, T, RANK>

fn default() -> Self

Returns the “default value” for a type.

impl<D: Float, const RANK: usize> Layer<SymTensor> for Tanhshrink<D, SymTensor, RANK>

type Output = SymTensor

This layer’s output type.

fn call(&self, input: SymTensor) -> SymTensor

Applies this layer to input.

impl<D: Float, T: Tensor<D, RANK> + EagerTensor, const RANK: usize> Layer<T> for Tanhshrink<D, T, RANK>

type Output = T

This layer’s output type.

fn call(&self, _input: T) -> Self::Output

Applies this layer to input.