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 = {})#