embedding_op

→TensorImplPtr

int embedding_op(const int & weight, const int & indices, int padding_idx)

Integer-index embedding lookup — free-function entry point.

Thin wrapper that delegates to EmbeddingBackward::forward. Used by the pybind11 binding layer and by other C++ call sites.

weightTensorImplPtr

Embedding matrix of shape (num_embeddings, embed_dim).

indicesTensorImplPtr

Integer-typed tensor of any shape.

padding_idxint

Index to mask; negative disables masking.

TensorImplPtr

Output of shape (*indices.shape, embed_dim).