Struct faiss::nn::Linear

struct Linear

minimal translation of nn.Linear

Public Functions

Linear(size_t in_features, size_t out_features, bool bias = true)
Tensor2D operator()(const Tensor2D &x) const

Public Members

size_t in_features
size_t out_features
std::vector<float> weight
std::vector<float> bias