in kbc/models.py [0:0]
def forward(self, x): lhs = self.lhs(x[:, 0]) rel = self.rel(x[:, 1]) rhs = self.rhs(x[:, 2]) return (lhs * rel) @ self.rhs.weight.t(), (lhs, rel, rhs)