torch.nn.functional.cosine_similarity

torch.nn.functional.cosine_similarity(x1, x2, dim=1, eps=1e-8) → Tensor

Возвращает косинусное сходство между x1 и x2, вычисленное вдоль dim. x1 и x2 должны быть совместимы по форме для трансляции. dim относится к размерности в этой общей форме. Размерность dim выходного тензора сжимается (см. torch.squeeze()), что приводит к тому, что выходной тензор имеет на одну размерность меньше.

$$\text{similarity} = \dfrac{x_1 \cdot x_2}{\max(\Vert x_1 \Vert _2, \epsilon) \cdot \max(\Vert x_2 \Vert _2, \epsilon)}$$

Поддерживает продвижение типа.

Parameters

• x1 (Tensor) – Первый входной тензор.
• x2 (Tensor) – Второй входной тензор.
• dim (int, optional) – Размерность, по которой вычисляется косинусное сходство. По умолчанию: 1
• eps (float, optional) – Малое значение для предотвращения деления на ноль. По умолчанию: 1e-8

Пример:

>>> input1 = torch.randn(100, 128)
>>> input2 = torch.randn(100, 128)
>>> output = F.cosine_similarity(input1, input2)
>>> print(output)