pub struct LayerNormConfig<D = Ix2> {
pub axis: Option<Axis>,
pub dim: D,
pub eps: f64,
}Fields§
§axis: Option<Axis>§dim: D§eps: f64Implementations§
Source§impl<D> Config<D>where
D: Dimension,
impl<D> Config<D>where
D: Dimension,
pub fn new() -> ConfigBuilder<D>
pub fn axis(&self) -> Option<&Axis>
pub fn axis_mut(&mut self) -> &mut Option<Axis>
pub fn eps(&self) -> f64
pub fn eps_mut(&mut self) -> &mut f64
pub fn dim(&self) -> D::Pattern
pub fn dim_mut(&mut self) -> &mut D
pub fn ndim(&self) -> usize
pub fn raw_dim(&self) -> D
pub fn shape(&self) -> &[usize]
pub fn shape_mut(&mut self) -> &mut [usize]
Trait Implementations§
Auto Trait Implementations§
impl<D> Freeze for Config<D>where
D: Freeze,
impl<D> RefUnwindSafe for Config<D>where
D: RefUnwindSafe,
impl<D> Send for Config<D>where
D: Send,
impl<D> Sync for Config<D>where
D: Sync,
impl<D> Unpin for Config<D>where
D: Unpin,
impl<D> UnwindSafe for Config<D>where
D: UnwindSafe,
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