How to designthe SAGAN self attention layer?

paper at: https://arxiv.org/abs/1805.08318

I'm curious if the self-attention layer described in this paper could be implemented in MATLAB. Would it require a custom layer to construct? Does not seem feasible inside the Deep Network Designer. I'd appreciate a pointer in the right direction.

Thank you!

[fneg]

Hi, have you been able to implement this attention block in matlab?

Answer 1

[Divya Gaddipati]

You would have to create a custom layer. If you are using R2020a version, you'll be able to load the custom layer into the Deep Network Designer from the workspace.

You can refer to the following link for information on creating a custom layer:

https://www.mathworks.com/help/deeplearning/ug/define-custom-deep-learning-layers.html