pub struct LayerNorm { /* private fields */ }
Expand description
Layer that applies layer normalization.
Implementations§
Source§impl LayerNorm
impl LayerNorm
Sourcepub fn new<'a>(
vs: impl Borrow<PathExt<'a>>,
normalized_shape: impl Into<Vec<i64>>,
eps: f64,
elementwise_affine: bool,
) -> Self
pub fn new<'a>( vs: impl Borrow<PathExt<'a>>, normalized_shape: impl Into<Vec<i64>>, eps: f64, elementwise_affine: bool, ) -> Self
Construct a layer normalization layer.
The mean and standard deviation are computed over the last
number of dimensions with the shape defined by
normalized_shape
. If elementwise_affine
is True
, a
learnable affine transformation of the shape
normalized_shape
is added after normalization.
Trait Implementations§
Auto Trait Implementations§
impl Freeze for LayerNorm
impl RefUnwindSafe for LayerNorm
impl Send for LayerNorm
impl !Sync for LayerNorm
impl Unpin for LayerNorm
impl UnwindSafe for LayerNorm
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<M> FallibleModuleT for Mwhere
M: FallibleModule,
impl<M> FallibleModuleT for Mwhere
M: FallibleModule,
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