Struct Normalize

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

Normalize the mean and variance of elements in a slice.

This normalizes elements according to the formula:

output[i] = (input[i] - pre_scale_bias) * scale * element_scale[i] + bias + element_bias[i]

Panics

Dispatching the operation panics if any of the slices have different lengths.

Implementations

impl<'src, 'dst> Normalize<'src, 'dst>

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

Create a normalize operation which reads input and writes the normalized output to output.

pub fn new_mut(input: &'dst mut [f32], opts: NormalizeOptions<'src>) -> Self

where 'dst: 'src,

Create a normalize operation which normalizes input in-place.

Trait Implementations

impl<'dst> SimdOp for Normalize<'_, '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.