recently I am learning NumPy\'s usage and I met a question which gives me 3 array:
- q: A numpy array of shape (1, K) (queries) - k: A numpy array of shape (N, K)