Function torch::nn::functional::embedding

• Defined in File embedding.h

Function Documentation

inline Tensor torch::nn::functional::embedding(const Tensor &input, const Tensor &weight, const EmbeddingFuncOptions &options = {})

See https://pytorch.org/docs/main/nn.functional.html#torch.nn.functional.embedding about the exact behavior of this functional.

See the documentation for torch::nn::functional::EmbeddingFuncOptions class to learn what optional arguments are supported for this functional.

Example:

namespace F = torch::nn::functional;
F::embedding(input, weight,
F::EmbeddingFuncOptions().norm_type(2.5).scale_grad_by_freq(true).sparse(true));