(self, embeds1, embeds2)
| 502 | return (self.featureExtract(embeds1) * self.featureExtract(embeds2)).sum(dim=-1) |
| 503 | |
| 504 | def crossPred(self, embeds1, embeds2): |
| 505 | return self.featureExtract(embeds1) @ self.featureExtract(embeds2).T |
| 506 | |
| 507 | |
| 508 |
no test coverage detected