Transformer Block

Trace attention, feed-forward layers, residual connections, and normalization through one transformer block with explicit tensor shapes.