结构体 faiss::nn::Embedding

struct Embedding

nn.Embedding 的最小化翻译

公共函数

Embedding(size_t num_embeddings, size_t embedding_dim)

Tensor2D operator()(const Int32Tensor2D&) const

inline float *data()

inline const float *data() const

公共成员

size_t num_embeddings

size_t embedding_dim

std::vector<float> weight