//clika-runtime/io.clika.runtime/Ops/diagEmbed

diagEmbed

[common]
fun diagEmbed(input: Tensor, offset: Long = 0, dim1: Long = -2L, dim2: Long = -1L): Tensor

diagEmbed(input: Tensor, offset: Long = 0L, dim1: Long = -2L, dim2: Long = -1L): the diag_embed operator. Embed the LAST dim along the diagonal of a fresh (dim1, dim2) plane: output rank = input rank + 1, both new dims sized last + |offset|, the inverse of diagonal. Defaults place the plane at the trailing two dims. Copies.