Rate this Page

Class EmbeddingBag#

Inheritance Relationships#

Base Type#

Class Documentation#

class EmbeddingBag : public torch::nn::ModuleHolder<EmbeddingBagImpl>#

A ModuleHolder subclass for EmbeddingBagImpl.

See the documentation for EmbeddingBagImpl class to learn what methods it provides, and examples of how to use EmbeddingBag with torch::nn::EmbeddingBagOptions. See the documentation for ModuleHolder to learn about PyTorch’s module storage semantics.

Public Static Functions

static inline EmbeddingBag from_pretrained(const torch::Tensor &embeddings, const EmbeddingBagFromPretrainedOptions &options = {})#

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