quaterion.distances.cosine 模块

class Cosine

基类：BaseDistance

计算余弦相似度（以及将其解释为距离）。

注意

此指标的输出范围为 0 -> 1。

static distance(x: Tensor, y: Tensor) → Tensor

计算距离，即值越低，样本越相似。

参数:

• x – 形状：(batch_size, embedding_dim)

• y – 形状：(batch_size, embedding_dim)

返回:

距离 - 形状 – (batch_size,)

static distance_matrix(x: Tensor, y: Tensor | None = None) → Tensor

计算距离矩阵，即 x 和 y 中所有可能对之间的距离。

参数:

• x – 形状：(batch_size, embedding_dim)

• y – 形状：(batch_size, embedding_dim)。如果 y is None，则将 x 赋值给 y。

返回:

距离矩阵 - 形状 – (batch_size, batch_size)

static similarity(x: Tensor, y: Tensor) → Tensor

计算相似度，即值越高，样本越相似。

参数:

• x – 形状：(batch_size, embedding_dim)

• y – 形状：(batch_size, embedding_dim)

返回:

相似度 - 形状 – (batch_size,)

static similarity_matrix(x: Tensor, y: Tensor | None = None) → Tensor

计算相似度矩阵，即 x 和 y 中所有可能对之间的相似度。

参数:

• x – 形状：(batch_size, embedding_dim)

• y – 形状：(batch_size, embedding_dim)。如果 y is None，则将 x 赋值给 y。

返回:

相似度矩阵 - 形状 – (batch_size, batch_size)