Class EmbeddingBag

Inheritance Relationships

Base Type

Class Documentation

class torch::nn::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

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.


Access comprehensive developer documentation for PyTorch

View Docs


Get in-depth tutorials for beginners and advanced developers

View Tutorials


Find development resources and get your questions answered

View Resources