(self,x, A)
| 12 | super(nconv,self).__init__() |
| 13 | |
| 14 | def forward(self,x, A): |
| 15 | # x.shape = (batch, dim, nodes, seq_len) |
| 16 | # A.shape = (node, node) |
| 17 | x = torch.einsum('ncwl,vw->ncvl', (x, A)) |
| 18 | return x.contiguous() |
| 19 | |
| 20 | |
| 21 | class wconv(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected