pytorch cuda 欧式距离 矩阵运算

输入为tensor矩阵向量

欧式距离

def get_elu_dis(data):
    return ((-2*(()))+((data),axis=1,keepdim=True)+((()),axis=0,keepdim=True))

余玄距离

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