pub struct Relu2;
Implementations§
Trait Implementations§
Source§impl Module for Relu2
impl Module for Relu2
type Input = Tensor
type Output = Tensor
fn forward(&self, input: &Self::Input) -> Self::Output
fn gather_params(&self, params: &mut HashMap<usize, Tensor>)
fn update_params(&self, params: &mut HashMap<usize, Tensor>)
fn gather_named_params( &self, prefix: &str, params: &mut HashMap<String, Tensor>, )
fn update_named_params( &self, prefix: &str, params: &mut HashMap<String, Tensor>, )
fn params(&self) -> HashMap<usize, Tensor>
fn named_params(&self, prefix: &str) -> HashMap<String, Tensor>
fn to_safetensors<P>(&self, filename: P)
fn update_by_safetensors<P>(&self, filenames: &[P], device: impl AsDevice)
impl Copy for Relu2
impl NonTrainableModule for Relu2
Auto Trait Implementations§
impl Freeze for Relu2
impl RefUnwindSafe for Relu2
impl Send for Relu2
impl Sync for Relu2
impl Unpin for Relu2
impl UnwindSafe for Relu2
Blanket Implementations§
Source§impl<T, M> ApplyModule<M> for Twhere
M: Module<Input = T>,
impl<T, M> ApplyModule<M> for Twhere
M: Module<Input = T>,
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> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
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