text: Title: Question about using different model for Score-CAM
  Body: Hi,
  
  I see you use `reshape_transform(tensor, height, width)` for different type of transformer model. But I still confused about how to use that.
  
  I want use Score-CAM for LeViT-256 and Nest-T and DeiT-S-distill.
  
  The example of reshape_transform you give didn't work for these models