(self, t, x)
| 359 | self.out = [] |
| 360 | |
| 361 | def forward(self, t, x): |
| 362 | adj = self.adj + torch.eye(self.adj.size(0)).to(x.device) |
| 363 | d = adj.sum(1) |
| 364 | _d = torch.diag(torch.pow(d, -0.5)) |
| 365 | adj_norm = torch.mm(torch.mm(_d, adj), _d) |
| 366 | |
| 367 | self.out.append(x) |
| 368 | self.nfe += 1 |
| 369 | ax = self.nconv(x, adj_norm) |
| 370 | f = 0.5 * self.alpha * (ax - x) |
| 371 | return f |
| 372 | |
| 373 | |
| 374 | class CGPODEBlock(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected