log_pos = torch.bmm(pos_embedding, input_embedding.unsqueeze(2)).squeeze() # 计算输入与负向标签间的夹角余弦值:用bmm函数完成两个带批次数据的矩阵相乘【bmm函数处理的必须是批次数据,即形状为{b,m,n]与[b,n,m]矩阵相乘;】 log_neg = torch.bmm(neg_embedding, -input_embedding.unsqueeze(2)).sq...