Struct LogSoftmax 

pub struct LogSoftmax<'src, 'dst> { /* private fields */ }

Computes the log softmax function over a slice of floats.

This is conceptually log(softmax(x)) which can be rewritten as log(exp(x) / sum(exp(x))).

Implementations

impl<'src, 'dst> LogSoftmax<'src, 'dst>

pub fn new(input: &'src [f32], output: &'dst mut [MaybeUninit<f32>]) -> Self

Construct a log softmax operation which reads input and writes to output.

pub fn new_mut(input: &'dst mut [f32]) -> Self

where 'dst: 'src,

Construct a log softmax operation which updates input in place.

Trait Implementations

impl<'dst> SimdOp for LogSoftmax<'_, 'dst>

type Output = &'dst mut [f32]

The normalized elements.

fn eval<I: Isa>(self, isa: I) -> Self::Output

Evaluate the operation using the given instruction set.

fn dispatch(self) -> Self::Output

where Self: Sized,

Dispatch this operation using the preferred ISA for the current platform.