Struct icu_segmenter::complex::lstm::matrix::MatrixBorrowedMut

pub(super) struct MatrixBorrowedMut<'a, const D: usize> {
    pub(super) data: &'a mut [f32],
    pub(super) dims: [usize; D],
}

A D-dimensional, mutably borrowed matrix.

Fields

data: &'a mut [f32]

dims: [usize; D]

Implementations

impl<'a> MatrixBorrowedMut<'a, 1>

pub(super) fn dim(&self) -> usize

impl<'a> MatrixBorrowedMut<'a, 2>

pub(super) fn dim(&self) -> (usize, usize)

impl<'a> MatrixBorrowedMut<'a, 3>

pub(super) fn dim(&self) -> (usize, usize, usize)

impl<'a, const D: usize> MatrixBorrowedMut<'a, D>

pub(super) fn as_borrowed(&self) -> MatrixBorrowed<'_, D>

pub(super) fn as_mut_slice(&mut self) -> &mut [f32]

pub(super) fn copy_submatrix<const M: usize>(&mut self, from: usize, to: usize)

pub(super) fn add(&mut self, other: MatrixZero<'_, D>) -> Option<()>

pub(super) fn softmax_transform(&mut self)

Mutates this matrix by applying a softmax transformation.

pub(super) fn sigmoid_transform(&mut self)

pub(super) fn tanh_transform(&mut self)

pub(super) fn convolve( &mut self, i: MatrixBorrowed<'_, D>, c: MatrixBorrowed<'_, D>, f: MatrixBorrowed<'_, D>, )

pub(super) fn mul_tanh( &mut self, o: MatrixBorrowed<'_, D>, c: MatrixBorrowed<'_, D>, )

impl<'a> MatrixBorrowedMut<'a, 1>

pub(super) fn add_dot_2d( &mut self, a: MatrixBorrowed<'_, 1>, b: MatrixZero<'_, 2>, )

Calculate the dot product of a and b, adding the result to self.

Note: For better dot product efficiency, if b is MxN, then a should be N; this is the opposite of standard practice.

impl<'a> MatrixBorrowedMut<'a, 2>

pub(super) fn add_dot_3d_1( &mut self, a: MatrixBorrowed<'_, 1>, b: MatrixZero<'_, 3>, )

Calculate the dot product of a and b, adding the result to self.

Self should be MxN; a, O; and b, MxNxO.

pub(super) fn add_dot_3d_2( &mut self, a: MatrixZero<'_, 1>, b: MatrixZero<'_, 3>, )

Calculate the dot product of a and b, adding the result to self.

Self should be MxN; a, O; and b, MxNxO.