Enum rust_bert::Activation
source · pub enum Activation {
gelu,
relu,
swish,
mish,
gelu_new,
tanh,
identity,
}
Variants§
gelu
Gaussian Error Linear Unit (Hendrycks et al., 2016,)
relu
Rectified Linear Unit
swish
Swish (Ramachandran, 2017)
mish
Mish (Misra, 2019)
gelu_new
Gaussian Error Linear Unit (New) (Hendrycks et al., 2016,)
tanh
Tanh
identity
Identity
Implementations§
source§impl Activation
impl Activation
pub fn get_function(&self) -> TensorFunction
Trait Implementations§
source§impl Clone for Activation
impl Clone for Activation
source§fn clone(&self) -> Activation
fn clone(&self) -> Activation
Returns a copy of the value. Read more
1.0.0 · source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source
. Read moresource§impl Debug for Activation
impl Debug for Activation
source§impl<'de> Deserialize<'de> for Activation
impl<'de> Deserialize<'de> for Activation
source§fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
Deserialize this value from the given Serde deserializer. Read more