Open dhananjaisharma10 opened 3 years ago
I would like to add support for the Attention Augmented Conv2d layer from the paper Attention Augmented Convolutional Networks.
This feature is useful for the following key reasons:
The feature would simply be a class inheriting torch.nn.Module.
torch.nn.Module
I look forward to hearing from the team. Thank you!
Any idea wether this was ever implemented?
🚀 Feature
I would like to add support for the Attention Augmented Conv2d layer from the paper Attention Augmented Convolutional Networks.
Motivation & Examples
This feature is useful for the following key reasons:
The feature would simply be a class inheriting
torch.nn.Module
.I look forward to hearing from the team. Thank you!