检索系统向量化计算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