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: AutodiffBackend> AutodiffModule<B> for Softplus
impl<B: AutodiffBackend> AutodiffModule<B> for Softplus
Source§type InnerModule = Softplus
type InnerModule = Softplus
Inner module without auto-differentiation.
Source§fn valid(&self) -> Self::InnerModule
fn valid(&self) -> Self::InnerModule
Get the same module, but on the inner backend without auto-differentiation.
Source§impl<B: Backend> Module<B> for Softplus
impl<B: Backend> Module<B> for Softplus
Source§type Record = ConstantRecord
type Record = ConstantRecord
Type to save and load the module.
Source§fn visit<V: ModuleVisitor<B>>(&self, _visitor: &mut V)
fn visit<V: ModuleVisitor<B>>(&self, _visitor: &mut V)
Visit each tensor parameter in the module with a visitor.
Source§fn map<M: ModuleMapper<B>>(self, _mapper: &mut M) -> Self
fn map<M: ModuleMapper<B>>(self, _mapper: &mut M) -> Self
Map each tensor parameter in the module with a mapper.
Source§fn load_record(self, _record: Self::Record) -> Self
fn load_record(self, _record: Self::Record) -> Self
Load the module state from a record.
Source§fn into_record(self) -> Self::Record
fn into_record(self) -> Self::Record
Convert the module into a record containing the state.
Source§fn to_device(self, _: &B::Device) -> Self
fn to_device(self, _: &B::Device) -> Self
Move the module and all of its sub-modules to the given device. Read more
Source§fn fork(self, _: &B::Device) -> Self
fn fork(self, _: &B::Device) -> Self
Fork the module and all of its sub-modules to the given device. Read more
Source§fn collect_devices(&self, devices: Devices<B>) -> Devices<B>
fn collect_devices(&self, devices: Devices<B>) -> Devices<B>
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>
Available on crate feature
std only.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>
Available on crate feature
std only.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