Class EmbeddingBag

• Defined in File embedding.h

Inheritance Relationships

Base Type

• public torch::nn::ModuleHolder< EmbeddingBagImpl > (Template Class ModuleHolder)

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.