argmin-math 0.5.1

argmin math abstractions
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

use crate::ArgminEye;
use faer::{Mat, Shape};
use faer_traits::ComplexField;

impl<E: ComplexField, R: Shape, C: Shape> ArgminEye for Mat<E, R, C>
where
    R: TryFrom<usize>,
    C: TryFrom<usize>,
{
    fn eye(n: usize) -> Self {
        let (nr, nc) = match (R::try_from(n), C::try_from(n)) {
            (Ok(nr), Ok(nc)) => (nr, nc),
            _ => panic!("invalid matrix size for index type"),
        };

        Mat::identity(nr, nc)
    }

    fn eye_like(&self) -> Self {
        let nr = self.nrows();
        let nc = self.ncols();
        //@note(geo-ant) in the nalgebra implementation we also enforce
        // that the matrix is square, which we don't enforce here
        Mat::identity(nr, nc)
    }
}