Nice work!And i reading your code recently. But i cannot understand well about the implementation of the mask in shifted window attention. I simply draw a picture like below. The red mean the mask,...