Skip to content
Permalink
Browse files

attn shape flatten

  • Loading branch information
bknyaz committed Jan 3, 2020
1 parent c84cee3 commit 328498186090c6b48d6442cd2ab02b079fc37ccc
Showing with 1 addition and 1 deletion.
  1. +1 −1 examples/triangles_sag_pool.py
@@ -13,7 +13,7 @@

class HandleNodeAttention(object):
def __call__(self, data):
data.attn = torch.softmax(data.x, dim=0)
data.attn = torch.softmax(data.x, dim=0).flatten()
data.x = None
return data

0 comments on commit 3284981

Please sign in to comment.
You can’t perform that action at this time.