How to make attention map for Swin Transformers?

Question

I have been working on attention maps feature for one of my projects. I have fine-tuned an image classification model for the same. So how can I develop attention maps for Swin Transformers using Pytorch ?

Thanks in advance !!