Class EmbeddingBag¶
Defined in File embedding.h
Page Contents
Inheritance Relationships¶
Base Type¶
public torch::nn::ModuleHolder< EmbeddingBagImpl >
(Template Class ModuleHolder)
Class Documentation¶
-
class EmbeddingBag : public torch::nn::ModuleHolder<EmbeddingBagImpl>¶
A
ModuleHolder
subclass forEmbeddingBagImpl
.See the documentation for
EmbeddingBagImpl
class to learn what methods it provides, and examples of how to useEmbeddingBag
withtorch::nn::EmbeddingBagOptions
. See the documentation forModuleHolder
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.
-
static inline EmbeddingBag from_pretrained(const torch::Tensor &embeddings, const EmbeddingBagFromPretrainedOptions &options = {})¶