Struct faiss::nn::Embedding

struct Embedding

minimal translation of nn.Embedding

Public Functions

Embedding(size_t num_embeddings, size_t embedding_dim)

Tensor2D operator()(const Int32Tensor2D&) const

inline float *data()

inline const float *data() const

Public Members

size_t num_embeddings

size_t embedding_dim

std::vector<float> weight