context_vector_2 = attention_weights_2 @ values
​
print(context_vector_2.shape)
print(context_vector_2)