search

utkuozbulak / pytorch-cnn-visualizations

Pytorch implementation of convolutional neural network visualization techniques
MIT License
7.81k stars 1.49k forks source link

attention #117

Closed yuan243212790 closed 1 year ago

yuan243212790 commented 1 year ago

Can it achieve attention to the heat map of the attention mechanism

utkuozbulak commented 1 year ago

It might but not natively. You will have to modify accordingly.