Skip to main content

//clika-runtime/io.clika.runtime/Ops/diag

diag

[common]
fun diag(input: Tensor, diagonal: Long = 0): Tensor

diag(input: Tensor, diagonal: Long = 0L): the diag operator. Rank-1 <-> rank-2 diagonal converter. A vector of length L becomes an (L + |diagonal|) x (L + |diagonal|) matrix carrying it on the chosen diagonal (zeros elsewhere); a matrix input reads the chosen diagonal back out as a vector. diagonal: 0 = main, +k above, -k below. Copies.