Struggling with vizualization of attention tensor

[Lhemamou]

Hi guys,

I have successfully trained a classifier with attention with context mecanism, but i 'm struggling with the way to call the function get_attention_tensor . Do you have any clues in order to make it work ?

Thanks !

Léo