tensorplay.nn.functional.cosine_similarity

tensorplay.nn.functional.cosine_similarity(x1: TensorBase, x2: TensorBase, dim: int = 1, eps: float = 1e-08) → TensorBase[source]

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