Sentence Embedding Not Averaged

def sentence_vector(tokens, embeddings):
    dim = len(embeddings[tokens[0]])
    total = [0] * dim
    for t in tokens:
        vec = embeddings[t]
        for i in range(dim):
            total[i] += vec[i]
    return total

emb = {'a': [2, 4], 'b': [4, 8]}
print(sentence_vector(['a', 'b'], emb))