《使用PyTorch 2.0加速Transformer:训练推理均拿下!》 torch.nn.functional.scaled_dot_product_attention 目前还不是稳定版本的API;