边感知注意力层

一篇题为《Efficient Edge-aware Attention Network for Graph Generation》的研究提出了一种边感知注意力层，通过头级分数调制注入结构信息，并用轻量残差机制更新边表示。

该边感知注意力层借助PyTorch FlexAttention实现，在保持生成质量的同时将峰值GPU显存降低约65%，并显著提升训练与采样吞吐量。