[docs]classToComplex(OpRun):op_domain="ai.onnx.complex"def_run(self,x):assertx.shape[-1]in(1,2,),f"Unexpected shape {x.shape}, it should a tensor (..., 2)"ifx.shape[-1]==1:return(x[...,0]+0j,)return(x[...,0]+1j*x[...,1],)
[docs]classComplexModule(OpRun):op_domain="ai.onnx.complex"def_run(self,x):assertx.dtypein(np.complex64,np.complex128,),f"Unexpected type {x.dtype}, it should a complex tensor"return(np.abs(x),)