La similitude cos (cosinus) est l'une des méthodes d'apprentissage du contraste telles que SimCLR qui est utilisée comme indice de similitude dans l'espace des fonctionnalités.
Comme il a été implémenté respectivement par Tensorflow et Pytorch, enregistrez-le sous forme de mémo. (Pour votre référence)
Pytorch
# input_la taille est(batchsize*Nombre de dimensions)
def cosine_matrix(a, b):
dot = torch.matmul(a, torch.t(b))
norm = torch.matmul(torch.norm(a, dim=1).unsqueeze(-1), torch.norm(b, dim=0).unsqueeze(0))
return dot / norm
Tensorflow
def cosine_matrix(a, b):
a_normed, _ = tf.linalg.normalize(a, axis=-1)
b_normed, _ = tf.linalg.normalize(b, axis=-1)
matrix = tf.matmul(a_normed, b_normed, transpose_b=True)
return matrix