CosineSimilarityΒΆ

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

Abstract Signature:

CosineSimilarity(x1: Tensor, x2: Tensor, dim: int = 1, eps: float = 1e-8)

PyTorch

API: torch.nn.functional.cosine_similarity
Strategy: Direct Mapping

JAX (Core)

API: optax.cosine_similarity
Strategy: Direct Mapping

Keras

API: keras.losses.cosine_similarity
Strategy: Direct Mapping

TensorFlow

API: tf.keras.losses.cosine_similarity
Strategy: Direct Mapping