(self, inputs)
| 19 | |
| 20 | |
| 21 | def forward(self, inputs): |
| 22 | real, imag = torch.chunk(inputs, 2,self.complex_axis) |
| 23 | real = self.r_prelu(real) |
| 24 | imag = self.i_prelu(imag) |
| 25 | return torch.cat([real,imag],self.complex_axis) |
| 26 | |
| 27 | class NavieComplexLSTM(nn.Module): |
| 28 | def __init__(self, input_size, hidden_size, projection_dim=None, bidirectional=False, batch_first=False): |
nothing calls this directly
no outgoing calls
no test coverage detected