pytorch cuda 欧式距离矩阵运算

最新推荐文章于 2024-09-04 16:34:01 发布

dsjdjsa

最新推荐文章于 2024-09-04 16:34:01 发布

阅读量3.2k

点赞数 2

本文链接：https://blog.csdn.net/qq_21046135/article/details/108684283

版权

输入为tensor矩阵向量

欧式距离

def get_elu_dis(data):
    return torch.sqrt((-2*data.mm(data.t()))+torch.sum(torch.square(data),axis=1,keepdim=True)+torch.sum(torch.square(data.t()),axis=0,keepdim=True))

余玄距离

def get_cos_dis(sentences_vec):
    sentences_vec = sentences_vec / torch.norm(sentences_vec, 2, 1, keepdim=True)
    sentences_vec_mat = F.relu(1 - torch.mm(sentences_vec, sentences_vec.t()), inplace=True).cpu().numpy()
    return sentences_vec_mat