How do I get attention map?

ml-jku / hopfield-layers

Hopfield Networks is All You Need

https://ml-jku.github.io/hopfield-layers/

Other

1.69k stars 189 forks source link

Open hoangphucITJP opened 2 years ago

hoangphucITJP commented 2 years ago

I'm trying to apply HopfieldPooling to the MIL problem. How do I get the attention map?