0ab6621dcd3e9cd97a257ea2d8d05b23bfba8609,openke/module/model/HolE.py,HolE,_calc,#HolE#,65

Before Change

	def _calc(self, head, tail, rel):
		relation_mention = tf.nn.l2_normalize(rel, 1)
		entity_mention = self._ccorr(head, tail)
		return -tf.sigmoid(tf.reduce_sum(relation_mention * entity_mention, 1, keep_dims = True))

After Change

		return self._real(res).flatten(start_dim = -2)

	def _calc(self, h, t, r, mode):
		if mode != "normal":
			h = h.view(-1, r.shape[0], h.shape[-1])
			t = t.view(-1, r.shape[0], t.shape[-1])
			r = r.view(-1, r.shape[0], r.shape[-1])
		score = self._ccorr(h, t) * r
		score = torch.sum(score, -1).flatten()
		return score

	def forward(self, data):

Instances