Tooltip text goes here.
[B, S, D] [B, S] [B, S, D] [B, S, D] [B, D] [B, D] Token Embeddings Attention Mask Expand Mask Multiply Sum Tokens L2 Normalize 1D Vector