Implement a correct multi-head self-attention forward pass with projections, reshaping, and masking.

Project Q, K, and V; reshape to heads; compute QKᵀ/√d; apply causal or padding masks before softmax; multiply by V; then concatenate heads and project. Test tensor shapes and fully masked rows.

Test empty input, one-element input, duplicates, boundary indices, invalid states, and the largest allowed size; state time and space complexity.