PyTorch 提供了 torch.nn.CosineEmbeddingLoss 类来实现余弦相似度损失函数。该类接受三个参数:输入向量 x1x1x1 和x2x2x2,以及目标标签 yyy(1 表示相似,-1 表示不相似)。损失函数的计算公式为: [ \text{loss}(x_1, x_2, y) = \begin{cases} 1 - \cos(x_1, x_2) & \text{if } y = 1 \ \...