//clika-runtime/io.clika.runtime/Ops/embedding

embedding

[common]
fun embedding(indices: Tensor, weight: Tensor, bias: Tensor? = null, activation: Activation? = null): Tensor

embedding(indices: Tensor, weight: Tensor, bias: Tensor? = null, activation: Activation? = null): the embedding operator. Embedding lookup: gathers rows of weight by indices, with an optional bias + activation epilogue.