W4 A1 | Transformers: scaled dot assessment

I miscalculated with dk = k.ndim, it should be seq_len_k