Module fast_transformers.clustering.hamming.cluster_cuda

Functions

def cluster(...)

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

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