quaterion.distances.base_distance 模块

class BaseDistance

基类: object

提供任何距离度量都应实现的基类。

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 为 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 为 None，则将 x 赋值给 y。

返回：

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