pub enum ActivationKind {
Tanh,
Relu,
Relu6,
Gelu,
Sigmoid,
Softplus,
Silu,
Linear,
}Expand description
Activation kinds recognized in DeePMD config (matches
get_activation_fn in utils/network.py).
Variants§
Tanh
Relu
Relu6
Gelu
Both gelu and gelu_tf map here — Python uses the tanh
approximation in both cases.
Sigmoid
Softplus
Silu
Linear
Implementations§
Trait Implementations§
Source§impl Clone for ActivationKind
impl Clone for ActivationKind
Source§fn clone(&self) -> ActivationKind
fn clone(&self) -> ActivationKind
Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · 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 ActivationKind
impl Debug for ActivationKind
Source§impl PartialEq for ActivationKind
impl PartialEq for ActivationKind
Source§fn eq(&self, other: &ActivationKind) -> bool
fn eq(&self, other: &ActivationKind) -> bool
Tests for
self and other values to be equal, and is used by ==.impl Copy for ActivationKind
impl Eq for ActivationKind
impl StructuralPartialEq for ActivationKind
Auto Trait Implementations§
impl Freeze for ActivationKind
impl RefUnwindSafe for ActivationKind
impl Send for ActivationKind
impl Sync for ActivationKind
impl Unpin for ActivationKind
impl UnsafeUnpin for ActivationKind
impl UnwindSafe for ActivationKind
Blanket Implementations§
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