I have figure out my problem. it is due to complex number. You can refer to Unexpected tensor scalar type with torch.onnx.export and ComplexFloat · Issue #59246 · pytorch/pytorch · GitHub 32.
The error log comes from pytorch/export.cpp at master · pytorch/pytorch · GitHub 3