inputs:             6 × 3
attention_scores:   6 × 6
attention_weights:  6 × 6
context_vectors:    6 × 3

        __
        __