Cosine_similarity — nnf_cosine_similarity • torch

Returns cosine similarity between x1 and x2, computed along dim.

Usage

nnf_cosine_similarity(x1, x2, dim = 2, eps = 1e-08)

Arguments

x1: (Tensor) First input.
x2: (Tensor) Second input (of size matching x1).
dim: (int, optional) Dimension of vectors. Default: 2
eps: (float, optional) Small value to avoid division by zero. Default: 1e-8

Details

$similarity = \frac{x_{1} \cdot x_{2}}{max (‖ x_{1} ‖_{2} \cdot ‖ x_{2} ‖_{2}, ϵ)}$