npair_loss

    参考阅读

    NPair损失需要成对的数据。NPair损失分为两部分：第一部分是对嵌入向量进行L2正则化；第二部分是每一对数据的相似性矩阵的每一行和映射到ont-hot之后的标签的交叉熵损失的和。

    • labels (Tensor) - 标签向量，形状为[batch_size]的1-D Tensor 。数据类型：float32、float64和int64。

    返回:

    经过npair loss计算之后的结果 Tensor 。

    代码示例：