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