mlx_rs::nn

Trait Activation

Source 
pub trait Activation:
    UnaryModule<Error = Exception>
    + Debug
    + DynClone { }
Expand description

A marker trait for activation functions used in transformers.

Implementors§

Source§

impl<M> Activation for M
where M: UnaryModule<Error = Exception> + Debug + DynClone,