Confused about the code based on the paper

[hayleeXinyi]

Thanks for your great work! It is useful and concise, we want to cite your paper. But I have some confusion when I read your paper. Why do you get average at line 91 of persam.py, seeing as how the paper calculates the sim map and then averages it?
target_embedding = target_feat.mean(0).unsqueeze(0)

[ArenaGrenade]

I think this is just vector math. F_I is taken outside the sum and what remains is the average of the target features they calculate in line 91. The code seems to be correct as far as this is concerned.