in kbc/models.py [0:0]
def score(self, x): lhs = self.lhs(x[:, 0]) rel = self.rel(x[:, 1]) rhs = self.rhs(x[:, 2]) return torch.sum(lhs * rel * rhs, 1, keepdim=True)