pub enum Activation {
Gelu,
NewGelu,
Relu,
Relu2,
Relu6,
Silu,
}
Variants§
Implementations§
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 Default for Activation
impl Default for Activation
Source§fn default() -> Activation
fn default() -> Activation
Returns the “default value” for a type. Read more
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
Source§impl Module for Activation
impl Module for Activation
type Input = Tensor
type Output = Tensor
fn forward(&self, input: &Self::Input) -> Self::Output
fn gather_params(&self, params: &mut HashMap<usize, Tensor>)
fn update_params(&self, params: &mut HashMap<usize, Tensor>)
fn gather_named_params( &self, prefix: &str, params: &mut HashMap<String, Tensor>, )
fn update_named_params( &self, prefix: &str, params: &mut HashMap<String, Tensor>, )
fn params(&self) -> HashMap<usize, Tensor>
fn named_params(&self, prefix: &str) -> HashMap<String, Tensor>
fn to_safetensors<P>(&self, filename: P)
fn update_by_safetensors<P>(&self, filenames: &[P], device: impl AsDevice)
Source§impl PartialEq for Activation
impl PartialEq for Activation
Source§impl ValueSpec for Activation
impl ValueSpec for Activation
impl Copy for Activation
impl StructuralPartialEq for Activation
impl TrainableModule for Activation
Auto Trait Implementations§
impl Freeze for Activation
impl RefUnwindSafe for Activation
impl Send for Activation
impl Sync for Activation
impl Unpin for Activation
impl UnwindSafe for Activation
Blanket Implementations§
Source§impl<T, M> ApplyModule<M> for Twhere
M: Module<Input = T>,
impl<T, M> ApplyModule<M> for Twhere
M: Module<Input = T>,
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self
into a Left
variant of Either<Self, Self>
if into_left
is true
.
Converts self
into a Right
variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self
into a Left
variant of Either<Self, Self>
if into_left(&self)
returns true
.
Converts self
into a Right
variant of Either<Self, Self>
otherwise. Read more