小丁同学

> Try to use the function hook with (register_forward_hook) to get the specific feature map. Thank you！

I have the same problem. Have you solve this?

> ```python > img = self.vit.to_patch_embedding(img) > img += self.vit.pos_embedding[:, :img.shape[1]] > img = self.vit.dropout(img) > img = self.vit.transformer(img) > ``` 谢谢！

Thank you very much for your patient response! Thank you!