Function neuronika::eye

pub fn eye(n: usize) -> Var<Input<Ix2>>

Creates a variable with an identity matrix of size n.

Panics

If n * n would overflow isize.

Examples

use neuronika;
use ndarray::Array2;

let tensor = neuronika::eye(3);
assert_eq!(*tensor.data(), Array2::eye(3));