结构体 faiss::nn::FFN

struct FFN

前馈层,扩展到隐藏维度,应用 ReLU 非线性,然后映射回原始维度

公共函数

inline FFN(int d, int h)
Tensor2D operator()(const Tensor2D &x) const

公共成员

Linear linear1
Linear linear2