identity
Creates an n x n square identity matrix with ones on the main diagonal and zeros elsewhere.
Signatures
inline fun <reified T : Any> Multik.identity(n: Int): D2Array<T>
fun <T> Multik.identity(n: Int, dtype: DataType): D2Array<T>
Parameters
Parameter | Type | Description |
|---|---|---|
|
| Number of rows and columns. |
|
| Element type (when not using reified generics). |
Returns: D2Array<T> — an n × n matrix with ones on the main diagonal and zeros elsewhere.
Example
val I = mk.identity<Double>(3)
// [[1.0, 0.0, 0.0],
// [0.0, 1.0, 0.0],
// [0.0, 0.0, 1.0]]
val intI = mk.identity<Int>(2, DataType.IntDataType)
// [[1, 0],
// [0, 1]]
28 February 2026