omega_2 = query_2 @ keys.T
print(omega_2)