Function complex_

tfjs-core/src/ops/complex.ts:47–57

* Converts two real numbers to a complex number. * * Given a tensor `real` representing the real part of a complex number, and a * tensor `imag` representing the imaginary part of a complex number, this * operation returns complex numbers elementwise of the form [r0, i0, r1, i1], * where r repr

(real: T|TensorLike, imag: T|TensorLike)

 * @doc {heading: 'Tensors', subheading: 'Creation'}
 */
function complex_<T extends Tensor>(real: T|TensorLike, imag: T|TensorLike): T {
  const $real = convertToTensor(real, 'real', 'complex');
  const $imag = convertToTensor(imag, 'imag', 'complex');
  util.assertShapesMatch(
      $real.shape, $imag.shape,
      `real and imag shapes, ${$real.shape} and ${$imag.shape}, ` +
          `must match in call to tf.complex().`);

  const inputs: ComplexInputs = {real: $real, imag: $imag};
  return ENGINE.runKernel(Complex, inputs as unknown as NamedTensorMap);
}

export const complex = /* @__PURE__ */ op({complex_});