Question About Predefined Concept Module

[jionie]

Hi, thanks for sharing this great work! I'm a little bit confusing about line 97 to line 101, why calculate hidden twice? What's the purpose of this part? Thanks!

95  hidden = torch.t(stock_to_concept).mm(x_hidden)

97  hidden = hidden[hidden.sum(1)!=0]
98  stock_to_concept = x_hidden.mm(torch.t(hidden))
99  # stock_to_concept = cal_cos_similarity(x_hidden, hidden)
100 stock_to_concept = self.softmax_s2t(stock_to_concept)
101 hidden = torch.t(stock_to_concept).mm(x_hidden)

103 concept_to_stock = cal_cos_similarity(x_hidden, hidden)