align/test.py [90:95]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        if criss is not None:
            semb = criss.embed(ss, langcode=configs.src_lang)
            temb = criss.embed(ts, langcode=configs.trg_lang)
            cos_matrix = cos(semb.unsqueeze(1), temb.unsqueeze(0)).unsqueeze(-1).unsqueeze(-1)
            ip_matrix = (semb.unsqueeze(1) * temb.unsqueeze(0)).sum(-1).unsqueeze(-1).unsqueeze(-1)
            feat_matrix = torch.cat((cos_matrix, ip_matrix), dim=-1)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



align/train.py [146:151]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            if criss is not None:
                semb = criss.embed(ss, langcode=configs.src_lang)
                temb = criss.embed(ts, langcode=configs.trg_lang)
                cos_matrix = cos(semb.unsqueeze(1), temb.unsqueeze(0)).unsqueeze(-1).unsqueeze(-1)
                ip_matrix = (semb.unsqueeze(1) * temb.unsqueeze(0)).sum(-1).unsqueeze(-1).unsqueeze(-1)
                feat_matrix = torch.cat((cos_matrix, ip_matrix), dim=-1)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -