(self, occ, prc)
| 60 | self.A = a_delta |
| 61 | |
| 62 | def forward(self, occ, prc): # occ.shape = [batch, node, seq] |
| 63 | x = torch.stack([occ, prc], dim=3) |
| 64 | x = self.encoder(x) |
| 65 | # l1 |
| 66 | x = self.gcn_l1(x) |
| 67 | x = torch.matmul(self.A, x) |
| 68 | x = self.act(x) |
| 69 | # l2 |
| 70 | x = self.gcn_l2(x) |
| 71 | x = torch.matmul(self.A, x) |
| 72 | x = self.act(x) |
| 73 | x = self.decoder(x) |
| 74 | return x |
| 75 | |
| 76 | |
| 77 | class LstmGcn(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected