(inputs, axis)
| 65 | self.real_lstm.flatten_parameters() |
| 66 | |
| 67 | def complex_cat(inputs, axis): |
| 68 | |
| 69 | real, imag = [],[] |
| 70 | for idx, data in enumerate(inputs): |
| 71 | r, i = torch.chunk(data,2,axis) |
| 72 | real.append(r) |
| 73 | imag.append(i) |
| 74 | real = torch.cat(real,axis) |
| 75 | imag = torch.cat(imag,axis) |
| 76 | outputs = torch.cat([real, imag],axis) |
| 77 | return outputs |
| 78 | |
| 79 | class ComplexConv2d(nn.Module): |
| 80 |