pub fn identity<T: HasAfEnum>(dims: Dim4) -> Array<T>
Create an identity array with 1’s in diagonal
dims
Identity matrix