How can I extract the embeddings in a dimension 1xdim?

Zian-Xu / Swin-MAE

Pytorch implementation of Swin MAE https://arxiv.org/abs/2212.13805

62 stars 12 forks source link

Open dnnxl opened 3 months ago

dnnxl commented 3 months ago

Hi thanks for the contribution, I would like to ask, how can I get the embeddings in a dimension of 1x dim?