您即将离开知乎，请注意您的账号和财产安全。
https://kkgithub.com/dhcode-cpp/NSA-pytorch/blob/main/native-sparse-attention-pytorch.ipynb
