(self, complex_axis=1)
| 12 | class cPReLU(nn.Module): |
| 13 | |
| 14 | def __init__(self, complex_axis=1): |
| 15 | super(cPReLU,self).__init__() |
| 16 | self.r_prelu = nn.PReLU() |
| 17 | self.i_prelu = nn.PReLU() |
| 18 | self.complex_axis = complex_axis |
| 19 | |
| 20 | |
| 21 | def forward(self, inputs): |