Week4, assignment, scaled_dot_product_attention()

Can someone explain the shape of q,v,k and the rationale please? thanks!

This description may be more helpful than the one in the course:

Thanks.