Struct faiss::nn::FFN

struct FFN

Feed forward layer that expands to a hidden dimension, applies a ReLU non linearity and maps back to the orignal dimension

Public Functions

inline FFN(int d, int h)

Tensor2D operator()(const Tensor2D &x) const

Public Members

Linear linear1

Linear linear2