All context vectors:
tensor([[0.4181, 0.4459, 0.6276],
        [0.4181, 0.4459, 0.6276],
        [0.4130, 0.5147, 0.5924],
        [0.4038, 0.5031, 0.5789],
        [0.4442, 0.4635, 0.5573],
        [0.3882, 0.5229, 0.5909]])

        __
        __