pub struct Softplus {
pub beta: f64,
}Expand description
Softplus layer.
Applies the softplus function element-wise:
softplus(x) = (1/beta) * log(1 + exp(beta * x))
Should be created with SoftplusConfig.
Fields§
§beta: f64The beta value.
Implementations§
Trait Implementations§
Source§impl<B> AutodiffModule<B> for Softpluswhere
B: AutodiffBackend,
impl<B> AutodiffModule<B> for Softpluswhere
B: AutodiffBackend,
Source§type InnerModule = Softplus
type InnerModule = Softplus
Inner module without auto-differentiation.
Source§fn valid(&self) -> <Softplus as AutodiffModule<B>>::InnerModule
fn valid(&self) -> <Softplus as AutodiffModule<B>>::InnerModule
Get the same module, but on the inner backend without auto-differentiation.
Source§impl<B> From<Softplus> for Activation<B>where
B: Backend,
impl<B> From<Softplus> for Activation<B>where
B: Backend,
Source§fn from(layer: Softplus) -> Activation<B>
fn from(layer: Softplus) -> Activation<B>
Converts to this type from the input type.
Source§impl<B> Module<B> for Softpluswhere
B: Backend,
impl<B> Module<B> for Softpluswhere
B: Backend,
Source§type Record = ConstantRecord
type Record = ConstantRecord
Type to save and load the module.
Source§fn visit<V>(&self, _visitor: &mut V)where
V: ModuleVisitor<B>,
fn visit<V>(&self, _visitor: &mut V)where
V: ModuleVisitor<B>,
Visit each tensor parameter in the module with a visitor.
Source§fn map<M>(self, _mapper: &mut M) -> Softpluswhere
M: ModuleMapper<B>,
fn map<M>(self, _mapper: &mut M) -> Softpluswhere
M: ModuleMapper<B>,
Map each tensor parameter in the module with a mapper.
Source§fn load_record(self, _record: <Softplus as Module<B>>::Record) -> Softplus
fn load_record(self, _record: <Softplus as Module<B>>::Record) -> Softplus
Load the module state from a record.
Source§fn into_record(self) -> <Softplus as Module<B>>::Record
fn into_record(self) -> <Softplus as Module<B>>::Record
Convert the module into a record containing the state.
Source§fn to_device(self, _: &<B as Backend>::Device) -> Softplus
fn to_device(self, _: &<B as Backend>::Device) -> Softplus
Move the module and all of its sub-modules to the given device. Read more
Source§fn fork(self, _: &<B as Backend>::Device) -> Softplus
fn fork(self, _: &<B as Backend>::Device) -> Softplus
Fork the module and all of its sub-modules to the given device. Read more
Source§fn collect_devices(
&self,
devices: Vec<<B as Backend>::Device>,
) -> Vec<<B as Backend>::Device>
fn collect_devices( &self, devices: Vec<<B as Backend>::Device>, ) -> Vec<<B as Backend>::Device>
Return all the devices found in the underneath module tree added to the given vector
without duplicates.
Source§fn devices(&self) -> Vec<<B as Backend>::Device>
fn devices(&self) -> Vec<<B as Backend>::Device>
Return all the devices found in the underneath module tree without duplicates.
Source§fn num_params(&self) -> usize
fn num_params(&self) -> usize
Get the number of parameters the module has, including all of its sub-modules.
Source§fn save_file<FR, PB>(
self,
file_path: PB,
recorder: &FR,
) -> Result<(), RecorderError>
fn save_file<FR, PB>( self, file_path: PB, recorder: &FR, ) -> Result<(), RecorderError>
Save the module to a file using the provided file recorder. Read more
Source§fn load_file<FR, PB>(
self,
file_path: PB,
recorder: &FR,
device: &<B as Backend>::Device,
) -> Result<Self, RecorderError>
fn load_file<FR, PB>( self, file_path: PB, recorder: &FR, device: &<B as Backend>::Device, ) -> Result<Self, RecorderError>
Load the module from a file using the provided file recorder. Read more
Source§fn quantize_weights(self, quantizer: &mut Quantizer) -> Self
fn quantize_weights(self, quantizer: &mut Quantizer) -> Self
Quantize the weights of the module.
Source§impl ModuleDisplay for Softplus
impl ModuleDisplay for Softplus
Source§fn custom_settings(&self) -> Option<DisplaySettings>
fn custom_settings(&self) -> Option<DisplaySettings>
Custom display settings for the module. Read more
Auto Trait Implementations§
impl Freeze for Softplus
impl RefUnwindSafe for Softplus
impl Send for Softplus
impl Sync for Softplus
impl Unpin for Softplus
impl UnwindSafe for Softplus
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