def cosine_sim()

in trainer/loss.py [0:0]


def cosine_sim(emb1, emb2):
    """compute cosine similarity of two embeddings
    Args:
        emb1 
        emb2 
    Returns:
        float: cosine similarity between (-1, 1)
    """    
    return emb1.mm(emb2.t())