Skip to content

.forward()

Look up embeddings for the given indices.

Parameters:

  • input (Tensor) –

    Integer tensor of indices with any shape S.

Returns:

  • Tensor

    Embedding tensor of shape (*S, embedding_dim).