检索系统向量化计算query-doc相似度

def cal_sim2(A,B):
    '''
    A :query [1,2]
    B: answers  [[1,2],[3,5]]
    '''
    need_norm=False
    A = np.array(A)
    B = np.array(B)
    A=A/np.linalg.norm(A)
    if need_norm:
        #A=A/np.linalg.norm(A)
        B= B/np.linalg.norm(B,axis=1).reshape(B.shape[0],1)

    sim = A.dot(B.T)
    return sim