Class EmbeddingModule

include/engine/framework/modules/lookup_modules.h:14–30

};

class EmbeddingModule {
public:
    explicit EmbeddingModule(EmbeddingConfig config);

    const EmbeddingConfig & config() const noexcept;
    const core::ModuleSchema & schema() const noexcept;

    core::TensorValue build(
        core::ModuleBuildContext & ctx,
        const core::TensorValue & indices,
        const core::TensorValue & weight) const;

    static const core::ModuleSchema & static_schema() noexcept;

private:
    EmbeddingConfig config_;
};

struct IndexedEmbeddingConfig {
    int64_t num_embeddings = 0;