Add support for Transformer models

[zhe-meng]

Hi, thanks for your work. Is it possible to add support for the transformer models, especifically layers implemented with the einops library.

[frgfm]

Hi @zhe-meng,

Thanks for your very relevant feature request!
I agree that support should be added to the library, I'll try to work on this soon. If you have some ideas about a PR, feel free to open one in the meantime!

[frgfm]

Hey @zhe-meng,

Small update, I started working on this and it will be handled in a few PRs:

• FLOPS estimation support (feat: Added FLOPS estimation support for nn.Transformer #48)
• MACs estimation support
• DMA estimation support
• Receptive field estimation support

I hope it will help 👍