pub struct ResidualBlock { /* private fields */ }Expand description
Residual block: runs input through a sequence of layers, then adds the
original input as a skip connection (output = layers(input) + input).
Implementations§
Source§impl ResidualBlock
impl ResidualBlock
Sourcepub fn new(layers: Vec<ModelLayer>) -> Self
pub fn new(layers: Vec<ModelLayer>) -> Self
Creates a new residual block wrapping the given layers.
Sourcepub fn layers(&self) -> &[ModelLayer]
pub fn layers(&self) -> &[ModelLayer]
Returns a reference to the inner layers.
Sourcepub fn forward_inference(&self, input: &Tensor) -> Result<Tensor, ModelError>
pub fn forward_inference(&self, input: &Tensor) -> Result<Tensor, ModelError>
Runs inference: passes input through all inner layers sequentially,
then adds the skip connection (output = layers_output + input).
Trait Implementations§
Source§impl Clone for ResidualBlock
impl Clone for ResidualBlock
Source§fn clone(&self) -> ResidualBlock
fn clone(&self) -> ResidualBlock
Returns a duplicate 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 moreAuto Trait Implementations§
impl Freeze for ResidualBlock
impl RefUnwindSafe for ResidualBlock
impl Send for ResidualBlock
impl Sync for ResidualBlock
impl Unpin for ResidualBlock
impl UnsafeUnpin for ResidualBlock
impl UnwindSafe for ResidualBlock
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> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
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