Skip to main content

identity

Function identity 

Source
pub fn identity<T: Zero + One, D0: Dim, D1: Dim>(n: usize) -> Array<T, (D0, D1)>
Expand description

Creates an identity matrix of size n x n.

ยงExamples

use mdarray::tensor;
use mdarray_linalg::utils::identity;

let i3 = identity::<f64, usize, usize>(3);
assert_eq!(i3, tensor![[1.,0.,0.],[0.,1.,0.],[0.,0.,1.]]);