pub struct Softmax {
pub dim: usize,
}
Fields§
§dim: usize
Implementations§
Trait Implementations§
Source§impl Primitive for Softmax
impl Primitive for Softmax
fn clone_boxed(&self) -> Box<dyn Primitive>
fn dot_label(&self) -> String
fn as_any(&self) -> &dyn Any
fn jvp( &self, output: &Tensor, _primals: &[Tensor], tangents: &[Tensor], ) -> Tensor
fn vjp( &self, output: &Tensor, _primals: &[Tensor], cotangent: &Tensor, ) -> Vec<Tensor>
impl Eq for Softmax
impl StructuralPartialEq for Softmax
Auto Trait Implementations§
impl Freeze for Softmax
impl RefUnwindSafe for Softmax
impl Send for Softmax
impl Sync for Softmax
impl Unpin for Softmax
impl UnwindSafe for Softmax
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