Just before the return, the output format is complex128,
but in the forloop during the training the output is automatically converted to stacked complex form. Just as below:
def forward(self, input): output=custom_function(input) print(output) return output
----main---- output=model.forward(input) print(output)
result: [a, b]
I don’t understand how this is happening and how to fix it.
Can I get some advice?
Thanks for reading!