如何将TensorFlow的实数转换为复数

tf.complex

complex( 
    real, 
    imag, 
    name=None
 )

定义在：tensorflow/python/ops/math_ops.py.

参考指南：数学函数>复数函数

将两个实数转换为复数.

给定 real 表示复数的实部的张量和 imag 表示复数的虚部的张量,该操作的返回形式为 \(a + bj \)的元数字的复数,其中 a 表示 real 部分,b 表示 imag 部分.

输入的张量 real 和 imag 必须具有相同的形状.

例如：

# 张量 'real' 是 [2.25, 3.25]
# 张量 `imag` 是 [4.75, 5.75]
tf.complex(real, imag) ==> [[2.25 + 4.75j], [3.25 + 5.75j]]

ARGS：

• real：张量.必须是以下类型之一：float32,float64.
• imag：张量.必须与 real 具有相同的类型.
• name：操作的名称(可选).

返回：

返回 complex64 或 complex128 类型的张量.