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