Optimize attention

[dholzmueller]

Hi, thanks for the nice repo! You can join the two column attentions like this: automl/nanoTabPFN#3

[jxucoder]

addressed. thx for the feedback

(using it in the comment for now. IMO, the 2 lines make it easier to understand the 2 attentions here)