torch_geometric.nn.pool.MIPSKNNIndex

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

Bases: KNNIndex

Performs fast \(k\)-nearest neighbor search (\(k\)-NN) based on the maximum inner product via the faiss library.

Parameters:

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