(self, occ, prc)
| 94 | self.A = a_delta |
| 95 | |
| 96 | def forward(self, occ, prc): # occ.shape = [batch, node, seq] |
| 97 | x = torch.stack([occ, prc], dim=3) |
| 98 | x = self.encoder(x) |
| 99 | x = torch.squeeze(x) |
| 100 | # l1 |
| 101 | x = self.gcn_l1(x) |
| 102 | x = torch.matmul(self.A, x) |
| 103 | x = self.act(x) |
| 104 | # l2 |
| 105 | x = self.gcn_l2(x) |
| 106 | x = torch.matmul(self.A, x) |
| 107 | x = self.act(x) |
| 108 | # lstm |
| 109 | x = x.transpose(1, 2) |
| 110 | x, _ = self.lstm(x) |
| 111 | x = x.transpose(1, 2) |
| 112 | x = self.decoder(x) |
| 113 | x = torch.squeeze(x) |
| 114 | return x |
| 115 | |
| 116 | |
| 117 | class LstmGat(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected