MCPcopy
hub / github.com/hunkim/PyTorchZeroToAll / get_att_score

Method get_att_score

seq2seq_models.py:141–143  ·  view source on GitHub ↗
(self, hidden, encoder_hidden)

Source from the content-addressed store, hash-verified

139 # score = h^T W h^e = h dot (W h^e)
140 # TODO: We need to implement different score models
141 def get_att_score(self, hidden, encoder_hidden):
142 score = self.attn(encoder_hidden)
143 return torch.dot(hidden.view(-1), score.view(-1))

Callers 1

get_att_weightMethod · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected