pub struct Linear { /* private fields */ }
Expand description
Linear (dense/fully-connected) layer.
Implementations§
Source§impl Linear
impl Linear
Sourcepub fn new(
name: &str,
weights: Box<dyn ArrayProtocol>,
bias: Option<Box<dyn ArrayProtocol>>,
activation: Option<ActivationFunc>,
) -> Self
pub fn new( name: &str, weights: Box<dyn ArrayProtocol>, bias: Option<Box<dyn ArrayProtocol>>, activation: Option<ActivationFunc>, ) -> Self
Create a new linear layer.
Sourcepub fn new_random(
name: &str,
in_features: usize,
out_features: usize,
withbias: bool,
activation: Option<ActivationFunc>,
) -> Self
pub fn new_random( name: &str, in_features: usize, out_features: usize, withbias: bool, activation: Option<ActivationFunc>, ) -> Self
Create a new linear layer with randomly initialized weights.
Trait Implementations§
Source§impl Layer for Linear
impl Layer for Linear
Source§fn layer_type(&self) -> &str
fn layer_type(&self) -> &str
Forward pass through the layer.
Get the layer type name for serialization.
fn forward( &self, inputs: &dyn ArrayProtocol, ) -> Result<Box<dyn ArrayProtocol>, OperationError>
Source§fn parameters(&self) -> Vec<Box<dyn ArrayProtocol>>
fn parameters(&self) -> Vec<Box<dyn ArrayProtocol>>
Get the layer’s parameters.
Source§fn parameters_mut(&mut self) -> Vec<&mut Box<dyn ArrayProtocol>>
fn parameters_mut(&mut self) -> Vec<&mut Box<dyn ArrayProtocol>>
Get mutable references to the layer’s parameters.
Source§fn update_parameter(
&mut self,
name: &str,
value: Box<dyn ArrayProtocol>,
) -> Result<(), OperationError>
fn update_parameter( &mut self, name: &str, value: Box<dyn ArrayProtocol>, ) -> Result<(), OperationError>
Update a specific parameter by name
Source§fn parameter_names(&self) -> Vec<String>
fn parameter_names(&self) -> Vec<String>
Get parameter names
Source§fn is_training(&self) -> bool
fn is_training(&self) -> bool
Check if the layer is in training mode.
Auto Trait Implementations§
impl Freeze for Linear
impl !RefUnwindSafe for Linear
impl Send for Linear
impl Sync for Linear
impl Unpin for Linear
impl !UnwindSafe for Linear
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
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