pub struct SoftmaxLayer;Expand description
Softmax layer over the last dimension.
Implementations§
Source§impl SoftmaxLayer
impl SoftmaxLayer
pub fn new() -> Self
pub fn forward_inference(&self, input: &Tensor) -> Result<Tensor, ModelError>
pub fn forward( &self, graph: &mut Graph, input: NodeId, ) -> Result<NodeId, ModelError>
Trait Implementations§
Source§impl Clone for SoftmaxLayer
impl Clone for SoftmaxLayer
Source§fn clone(&self) -> SoftmaxLayer
fn clone(&self) -> SoftmaxLayer
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 moreSource§impl Debug for SoftmaxLayer
impl Debug for SoftmaxLayer
Source§impl Default for SoftmaxLayer
impl Default for SoftmaxLayer
Source§fn default() -> SoftmaxLayer
fn default() -> SoftmaxLayer
Returns the “default value” for a type. Read more
Source§impl PartialEq for SoftmaxLayer
impl PartialEq for SoftmaxLayer
impl Copy for SoftmaxLayer
impl Eq for SoftmaxLayer
impl StructuralPartialEq for SoftmaxLayer
Auto Trait Implementations§
impl Freeze for SoftmaxLayer
impl RefUnwindSafe for SoftmaxLayer
impl Send for SoftmaxLayer
impl Sync for SoftmaxLayer
impl Unpin for SoftmaxLayer
impl UnsafeUnpin for SoftmaxLayer
impl UnwindSafe for SoftmaxLayer
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