def multiseq_flops(V, D):
    # (total coarse vocabulary) * (dims per coarse split) * (flops per squared distance)
    return (2 * V) * (D / 2) * 2
Given the number of coarse clusters and the dimension of the data,
dataset/ETHPy150Open yahoo/lopq/scripts/query_runtime.py/multiseq_flops