number of tokens × embedding dimensions
6 × 3

        __
        __