quaterion.loss.pairwise_loss 模块¶
- 类 PairwiseLoss(distance_metric_name: Distance = Distance.COSINE)[源代码]¶
基类:
SimilarityLoss成对损失的基类。
- 参数:
distance_metric_name – 距离函数的名称，例如
Distance。
- forward(embeddings: Tensor, pairs: Tensor, labels: Tensor, subgroups: Tensor) Tensor[源代码]¶
计算损失值。
- 参数:
embeddings – 形状: (batch_size, vector_length)
pairs – 形状: (2 * pairs_count,) - 包含 batch 中已知相似性对的列表
labels – 形状: (pairs_count,) - 对的相似性
subgroups – 形状: (2 * pairs_count,) - 对象的子组 ID
- 返回:
Tensor – 零尺寸 tensor，损失值
- training: bool¶
