https://mer.dev/posts/swin-transformer-hierarchical-vision-transformer-using-shifted-windows/