Module fast_transformers.clustering.hamming.cluster_cpu

Functions

def cluster(...)

cluster(arg0: at::Tensor, arg1: at::Tensor, arg2: at::Tensor, arg3: at::Tensor, arg4: at::Tensor, arg5: int, arg6: int) -> None

Cluster the hashed vectors by performing a few iterations of k-means