torch_geometric.nn.pool.MIPSKNNIndex

class MIPSKNNIndex(emb: Optional[Tensor] = None)

Bases: KNNIndex

基于最大内积通过faiss库执行快速的\(k\)-最近邻搜索（\(k\)-NN）。

Parameters:

emb (torch.Tensor, optional) – The data points to add. (default: None)