(self, x, W)
| 23 | super(wconv, self).__init__() |
| 24 | |
| 25 | def forward(self, x, W): |
| 26 | # x.shape = (batch, dim, nodes, seq_len) |
| 27 | # w.shape = (dim, dim) |
| 28 | x = torch.einsum('ncwl,vc->nvwl', (x, W)) |
| 29 | return x.contiguous() |
| 30 | |
| 31 | |
| 32 | class dy_nconv(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected