Attention Layer

Team,

Is there any example available in DL4J for Attention Layer (SelfAttentionLayer class)?

Regards

2 Likes

I would also be very much interested in such an example.

2 Likes

At the moment there are no examples that use it.

Would als be interested in seeing Attention NNs working with DL4J, especially given their recent popularity.

1 Like

Yea, I’d like to see an example also, especially how to make the example of the paper ‘Attention is All You Need’. imo, it would be very to see how to implement this model: https://ricardokleinklein.github.io/images/transformer/transformer.png .

1 Like